Skip to content

Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.

License

Notifications You must be signed in to change notification settings

pytorch-labs/tritonbench

This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Folders and files

NameName
Last commit message
Last commit date

Latest commit

10b4b4a · Dec 19, 2024
Dec 19, 2024
Dec 4, 2024
Dec 5, 2024
Nov 21, 2024
Dec 9, 2024
Dec 10, 2024
Dec 10, 2024
Dec 10, 2024
Dec 19, 2024
Dec 7, 2024
Nov 20, 2024
Oct 19, 2024
Oct 19, 2024
Oct 9, 2024
Nov 27, 2024
Dec 19, 2024
Nov 27, 2024
Nov 11, 2024
Dec 13, 2024

Repository files navigation

TritonBench

TritonBench is a collection of PyTorch operators used to evaluation the performance of Triton, and its integration with PyTorch.

Installation

The benchmark suite should be self-contained of its dependencies. To install, follow the steps below.

Step 1: clone the repository and checkout all submodules

$ git clone https://github.com/pytorch-labs/tritonbench.git
$ git submodule update --init --recursive

Step 2: run install.py

$ python install.py

By default, it will install the latest PyTorch nightly release and use the Triton version bundled with it.

Basic Usage

To benchmark an operator, run the following command:

$ python run.py --op gemm

Install as a library

To install as a library:

$ pip install -e .
# in your own benchmark script
import tritonbench
from tritonbench.utils import parser
op_args = parser.parse_args()
addmm_bench = tritonbench.load_opbench_by_name("addmm")(op_args)
addmm_bench.run()

Submodules

We depend on the following projects as a source of customized Triton or CUTLASS kernels:

License

TritonBench is BSD 3-Clause licensed, as found in the LICENSE file.

About

Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.

Resources

License

Code of conduct

Stars

Watchers

Forks

Releases

No releases published

Languages