Skip to content

Commit 6eba5c9

Browse files
committed
[Benchmarks] add SubmitKernel run with MeasureCompletion=1
1 parent a03da6d commit 6eba5c9

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

scripts/benchmarks/benches/compute.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -75,8 +75,9 @@ def benchmarks(self) -> list[Benchmark]:
7575

7676
if options.ur is not None:
7777
benches += [
78-
SubmitKernelUR(self, 0),
79-
SubmitKernelUR(self, 1),
78+
SubmitKernelUR(self, 0, 0),
79+
SubmitKernelUR(self, 1, 0),
80+
SubmitKernelUR(self, 1, 1),
8081
]
8182

8283
return benches
@@ -177,13 +178,14 @@ def bin_args(self) -> list[str]:
177178
]
178179

179180
class SubmitKernelUR(ComputeBenchmark):
180-
def __init__(self, bench, ioq):
181+
def __init__(self, bench, ioq, measureCompletion):
181182
self.ioq = ioq
183+
self.measureCompletion = measureCompletion
182184
super().__init__(bench, "api_overhead_benchmark_ur", "SubmitKernel")
183185

184186
def name(self):
185187
order = "in order" if self.ioq else "out of order"
186-
return f"api_overhead_benchmark_ur SubmitKernel {order}"
188+
return f"api_overhead_benchmark_ur SubmitKernel {order}" + (" with measure completion" if self.measureCompletion else "")
187189

188190
def explicit_group(self):
189191
return "SubmitKernel"
@@ -192,7 +194,7 @@ def bin_args(self) -> list[str]:
192194
return [
193195
f"--Ioq={self.ioq}",
194196
"--DiscardEvents=0",
195-
"--MeasureCompletion=0",
197+
f"--MeasureCompletion={self.measureCompletion}",
196198
"--iterations=100000",
197199
"--Profiling=0",
198200
"--NumKernels=10",

0 commit comments

Comments
 (0)