tree: 6d218481cbd752e6ba24d3c29f5b582b49fdd011 [path history] [tgz]
  1. python/
  2. CMakeLists.txt
  3. README.md
benchmark/README.md

Run benchmarks

To run all benchmarks:

cmake --build build --target benchmark-mpact

To run selected benchmarks, build the benchmark modules first:

cmake --build build --target build-benchmark-mpact

And then run the benchmark file:

python path/to/the/_benchmark.py

If you would like to run selected kernels in kernels_benchmark.py, you can use --benchmark-filter flag like the following example:

python path/to/the/kernels_benchmark.py --benchmark-filter=add

Profiler

Utils for profiling python scripts and pytorch models could be found in benchmark/python/utils/profiler.py.

To run the profiling example, use the following command:

python benchmark/python/utils/profiler.py