lightning-thunder icon indicating copy to clipboard operation
lightning-thunder copied to clipboard

benchmarking — create a notebook showing how to work with the single gpu benchmarks

Open mruberry opened this issue 1 year ago • 3 comments

The notebook should

  • show how to understand what the benchmark's options are
  • show how to programmatically run the benchmark
  • show how to create the callable that the benchmark is invoking
  • show how to acquire the input generator the benchmark is using to create the inputs to the callable

This way practitioners will be able to run the benchmark callable in their own contexts, which should support profiling.

fyi @kevinstephano — is this what you were thinking of when talking about extracting a callable from a benchmark?

cc @borda @apaz-cli @crcrpar

mruberry avatar Apr 16 '24 20:04 mruberry

This sounds good!

This might already be in mind with these bullet points but it would be helpful to show explicitly how to:

  • Show the trace
  • Execute specific fusions
  • Dump the debug for specific fusions

kevinstephano avatar Apr 22 '24 22:04 kevinstephano

fyi @riccardofelluga, who's thinking about benchmarking

mruberry avatar Apr 23 '24 14:04 mruberry

Regarding the debug for a specific fusion I've created #387. If that goes through I will show in this notebook how to use that information to dump it for specific fusions so that then it's just a matter of reading the output and creating the repro script

riccardofelluga avatar May 08 '24 19:05 riccardofelluga