Skip to content

Commit 5e4e39c

Browse files
Fix warmup in UR impl of SubmitKernel benchmark
Signed-off-by: Sergei Vinogradov <[email protected]>
1 parent b52da3b commit 5e4e39c

File tree

1 file changed

+16
-18
lines changed

1 file changed

+16
-18
lines changed

source/benchmarks/api_overhead_benchmark/implementations/ur/submit_kernel_ur.cpp

+16-18
Original file line numberDiff line numberDiff line change
@@ -65,26 +65,24 @@ static TestResult run(const SubmitKernelArguments &arguments, Statistics &statis
6565
std::vector<ur_event_handle_t> events(arguments.numKernels);
6666

6767
// warmup
68-
for (auto i = 0u; i < arguments.iterations; i++) {
69-
for (auto iteration = 0u; iteration < arguments.numKernels; iteration++) {
70-
EXPECT_UR_RESULT_SUCCESS(urKernelSetArgValue(
71-
kernel, 0, sizeof(int), nullptr,
72-
reinterpret_cast<void *>(&kernelExecutionTime)));
73-
74-
ur_event_handle_t *signalEvent = nullptr;
75-
if (arguments.useEvents) {
76-
signalEvent = &events[iteration];
77-
}
78-
79-
EXPECT_UR_RESULT_SUCCESS(urEnqueueKernelLaunch(
80-
queue, kernel, n_dimensions, nullptr, &local_size[0],
81-
&global_size[0], 0, nullptr, signalEvent));
68+
for (auto iteration = 0u; iteration < arguments.numKernels; iteration++) {
69+
EXPECT_UR_RESULT_SUCCESS(urKernelSetArgValue(
70+
kernel, 0, sizeof(int), nullptr,
71+
reinterpret_cast<void *>(&kernelExecutionTime)));
72+
73+
ur_event_handle_t *signalEvent = nullptr;
74+
if (arguments.useEvents) {
75+
signalEvent = &events[iteration];
8276
}
8377

84-
for (auto &event : events) {
85-
if (event)
86-
urEventRelease(event);
87-
}
78+
EXPECT_UR_RESULT_SUCCESS(urEnqueueKernelLaunch(
79+
queue, kernel, n_dimensions, nullptr, &local_size[0],
80+
&global_size[0], 0, nullptr, signalEvent));
81+
}
82+
83+
for (auto &event : events) {
84+
if (event)
85+
urEventRelease(event);
8886
}
8987

9088
// Benchmark

0 commit comments

Comments
 (0)