@@ -75,8 +75,9 @@ def benchmarks(self) -> list[Benchmark]:
75
75
76
76
if options .ur is not None :
77
77
benches += [
78
- SubmitKernelUR (self , 0 ),
79
- SubmitKernelUR (self , 1 ),
78
+ SubmitKernelUR (self , 0 , 0 ),
79
+ SubmitKernelUR (self , 1 , 0 ),
80
+ SubmitKernelUR (self , 1 , 1 ),
80
81
]
81
82
82
83
return benches
@@ -177,13 +178,14 @@ def bin_args(self) -> list[str]:
177
178
]
178
179
179
180
class SubmitKernelUR (ComputeBenchmark ):
180
- def __init__ (self , bench , ioq ):
181
+ def __init__ (self , bench , ioq , measureCompletion ):
181
182
self .ioq = ioq
183
+ self .measureCompletion = measureCompletion
182
184
super ().__init__ (bench , "api_overhead_benchmark_ur" , "SubmitKernel" )
183
185
184
186
def name (self ):
185
187
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 "" )
187
189
188
190
def explicit_group (self ):
189
191
return "SubmitKernel"
@@ -192,7 +194,7 @@ def bin_args(self) -> list[str]:
192
194
return [
193
195
f"--Ioq={ self .ioq } " ,
194
196
"--DiscardEvents=0" ,
195
- "--MeasureCompletion=0 " ,
197
+ f "--MeasureCompletion={ self . measureCompletion } " ,
196
198
"--iterations=100000" ,
197
199
"--Profiling=0" ,
198
200
"--NumKernels=10" ,
0 commit comments