Skip to content

Commit 18e482e

Browse files
Add options to test on L0v2
1 parent 1aa9202 commit 18e482e

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

sycl/test-e2e/format.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -343,6 +343,12 @@ def get_extra_env(sycl_devices):
343343
expanded = "env"
344344

345345
extra_env = get_extra_env([parsed_dev_name])
346+
backend, device = parsed_dev_name.split(":", 1)
347+
device_selector = parsed_dev_name
348+
if backend == "level_zero" and device.isdigit():
349+
extra_env.append(f"ZE_AFFINITY_MASK={device}")
350+
device_selector = f"{backend}:0"
351+
346352
if extra_env:
347353
expanded += " {}".format(" ".join(extra_env))
348354

sycl/test-e2e/lit.cfg.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -707,7 +707,7 @@ def remove_level_zero_suffix(devices):
707707
available_devices = {
708708
"opencl": ("cpu", "gpu", "fpga"),
709709
"cuda": "gpu",
710-
"level_zero": "gpu",
710+
"level_zero": ("gpu", "0", "1"),
711711
"hip": "gpu",
712712
"native_cpu": "cpu",
713713
"offload": "gpu",
@@ -1124,6 +1124,7 @@ def get_sycl_ls_verbose(sycl_device, env):
11241124
features.update(sg_size_features)
11251125
features.update(architecture_feature)
11261126
features.update(device_family)
1127+
features.update(aspects)
11271128

11281129
be, dev = sycl_device.split(":")
11291130
if dev.isdigit():

0 commit comments

Comments
 (0)