triton.testing

基准测试

这个类被 perf_report 函数用来以简洁的 API 生成折线图。

do_bench

对提供的函数的运行时进行基准测试。

do_bench_cudagraph

对提供的函数的运行时进行基准测试。

perf_report

将函数标记为进行基准测试。

assert_close

断言两个输入在一定容差范围内是接近的。