Add a Jupyter cell magic that wraps a cell with `cupyx.profiler.benchmark`

Open leofang opened this issue 1 month ago • 4 comments

As title. The %timeit magic is very useful but it is CPU-only, and often causes confusions for first-time GPU users programming in an IPython or Jupyter environment. Based on the teaching experience @brycelelbach suggested offline that we could add a Jupyter magic like, say, %gpu_timeit, that wraps a cell with cupyx.profiler.benchmark for GPU timing purposes.

Oct 31 '25 20:10 leofang

hey @leofang can i work on this?

Dec 05 '25 16:12 aman-coder03

Yes of course. Feel free to give it a shot.

Dec 05 '25 21:12 leofang

thanks!! @leofang

Dec 06 '25 05:12 aman-coder03

@leofang i have opened a PR, it would be really helpful if you take a look!!

Dec 07 '25 16:12 aman-coder03