apply comments

Author: mateuszpn

sycl/test-e2e/format.py

@@ -337,6 +337,8 @@ def get_extra_env(sycl_devices):
                 backend, device = parsed_dev_name.split(":", 1)
                 device_selector = parsed_dev_name
                 if backend == "level_zero" and device.isdigit():
+                    # level_zero:0 maps to a selector for the first GPU, so pin
+                    # execution to the requested device via ZE_AFFINITY_MASK.
                     extra_env.append(f"ZE_AFFINITY_MASK={device}")
                     device_selector = f"{backend}:0"
 

sycl/test-e2e/lit.cfg.py

@@ -706,7 +706,7 @@ def remove_level_zero_suffix(devices):
 available_devices = {
     "opencl": ("cpu", "gpu", "fpga"),
     "cuda": "gpu",
-    "level_zero": ("gpu", "0", "1"),
+    "level_zero": "gpu",
     "hip": "gpu",
     "native_cpu": "cpu",
     "offload": "gpu",