@@ -78,8 +78,9 @@ def benchmarks(self) -> list[Benchmark]:
78
78
79
79
if options .ur is not None :
80
80
benches += [
81
- SubmitKernelUR (self , 0 ),
82
- SubmitKernelUR (self , 1 ),
81
+ SubmitKernelUR (self , 0 , 0 ),
82
+ SubmitKernelUR (self , 1 , 0 ),
83
+ SubmitKernelUR (self , 1 , 1 ),
83
84
]
84
85
85
86
return benches
@@ -180,13 +181,14 @@ def bin_args(self) -> list[str]:
180
181
]
181
182
182
183
class SubmitKernelUR (ComputeBenchmark ):
183
- def __init__ (self , bench , ioq ):
184
+ def __init__ (self , bench , ioq , measureCompletion ):
184
185
self .ioq = ioq
186
+ self .measureCompletion = measureCompletion
185
187
super ().__init__ (bench , "api_overhead_benchmark_ur" , "SubmitKernel" )
186
188
187
189
def name (self ):
188
190
order = "in order" if self .ioq else "out of order"
189
- return f"api_overhead_benchmark_ur SubmitKernel { order } "
191
+ return f"api_overhead_benchmark_ur SubmitKernel { order } " + ( " with measure completion" if self . measureCompletion else "" )
190
192
191
193
def explicit_group (self ):
192
194
return "SubmitKernel"
@@ -195,7 +197,7 @@ def bin_args(self) -> list[str]:
195
197
return [
196
198
f"--Ioq={ self .ioq } " ,
197
199
"--DiscardEvents=0" ,
198
- "--MeasureCompletion=0 " ,
200
+ f "--MeasureCompletion={ self . measureCompletion } " ,
199
201
"--iterations=100000" ,
200
202
"--Profiling=0" ,
201
203
"--NumKernels=10" ,
0 commit comments