Skip to content

Commit 9cfab75

Browse files
committed
add nsys_gpu_speedup to BenchmarkOperatorMetrics
1 parent 0a498cc commit 9cfab75

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tritonbench/utils/triton_op.py

+2
Original file line numberDiff line numberDiff line change
@@ -226,6 +226,8 @@ class BenchmarkOperatorMetrics:
226226
mem_footprint_compression_ratio: Optional[float] = None
227227
# gbps
228228
gbps: Optional[float] = None
229+
# speedup for the summary of kernel GPU time only
230+
nsys_gpu_speedup: Optional[float] = None
229231

230232

231233
BUILTIN_METRICS = {x.name for x in fields(BenchmarkOperatorMetrics)} - {"extra_metrics"}

0 commit comments

Comments
 (0)