Skip to content

tensor-compiler/array-programming-benchmarks

Folders and files

NameName
Last commit message
Last commit date

Latest commit

759585b · Jun 22, 2021
Apr 10, 2021
Apr 25, 2021
Jun 22, 2021
Mar 11, 2021
Jun 22, 2021
Jun 22, 2021
Mar 8, 2021
Mar 10, 2021
Feb 12, 2021
Mar 18, 2021
Feb 13, 2021
Feb 15, 2021
Apr 1, 2021
Mar 24, 2021

Repository files navigation

Setup

Create a virtualenv (python3 -m venv venv) and activate it. Install requirements with pip install -r requirements.txt. Add benchmarks as python functions prefixed with test_ and have the function take in the benchmark fixture. An example is in numpy/windowing.py.

To run the benchmarks, use make python-bench to run all python benchmarks. Use the BENCHES flag to choose a particular set of benchmarks to run.

To write C++ benchmarks, we use the google/benchmark repository. Look at the example of taco/windowing.cpp for an example benchmark. Run make taco-bench to compile the taco benchmarks.