Skip to content

Commit 5c514d1

Browse files
[SYCL][Benchmark] Enable computeBench for cuda (#18084)
Enable computeBench to run on sycl with cuda backend via the benchmarking scripts.
1 parent 3a6c8de commit 5c514d1

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

devops/scripts/benchmarks/benches/compute.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,6 +70,9 @@ def setup(self):
7070
f"-DALLOW_WARNINGS=ON",
7171
]
7272

73+
if options.ur_adapter == "cuda":
74+
configure_command += ["-DBUILD_SYCL_WITH_CUDA=ON"]
75+
7376
if options.ur is not None:
7477
configure_command += [
7578
f"-DBUILD_UR=ON",
@@ -112,13 +115,17 @@ def enabled_runtimes(self, supported_runtimes=None):
112115
if options.ur is None:
113116
runtimes = [r for r in runtimes if r != RUNTIMES.UR]
114117

118+
# Filter out L0 if cuda backend
119+
if options.ur_adapter == "cuda":
120+
runtimes = [r for r in runtimes if r != RUNTIMES.LEVEL_ZERO]
121+
115122
return runtimes
116123

117124
def benchmarks(self) -> list[Benchmark]:
118125
if options.sycl is None:
119126
return []
120127

121-
if options.ur_adapter == "cuda" or options.ur_adapter == "hip":
128+
if options.ur_adapter == "hip":
122129
return []
123130

124131
benches = []

0 commit comments

Comments
 (0)