triton.testing

Benchmark

This class is used by the perf_report function to generate line plots with a concise API.

do_bench

Benchmark the runtime of the provided function.

do_bench_proton

Benchmark the runtime of kernels invoked by the provided function using the Proton profiler.

do_bench_cudagraph

Benchmark the runtime of the provided function.

do_bench_cudagraph_proton

Benchmark the runtime of kernels invoked by the provided function using the Proton profiler and CUDA graphs.

perf_report

Mark a function for benchmarking.

assert_close

Asserts that two inputs are close within a certain tolerance.