diff --git a/sycl/test-e2e/format.py b/sycl/test-e2e/format.py index b503108f937b..27064a73f648 100644 --- a/sycl/test-e2e/format.py +++ b/sycl/test-e2e/format.py @@ -353,6 +353,7 @@ def get_extra_env(sycl_devices): "linux", "windows", "preview-breaking-changes-supported", + "gpu", ]: if cond_features in test.config.available_features: conditions[cond_features] = True diff --git a/sycl/test-e2e/lit.cfg.py b/sycl/test-e2e/lit.cfg.py index d39cd7134e60..693388e62f47 100644 --- a/sycl/test-e2e/lit.cfg.py +++ b/sycl/test-e2e/lit.cfg.py @@ -917,12 +917,14 @@ def get_sycl_ls_verbose(sycl_device, env): env = copy.copy(llvm_config.config.environment) + backend_for_selector = backend.replace("_v2", "").replace("_v1", "") + # Find all available devices under the backend - env["ONEAPI_DEVICE_SELECTOR"] = backend + ":*" + env["ONEAPI_DEVICE_SELECTOR"] = backend_for_selector + ":*" detected_architectures = [] - platform_devices = remove_level_zero_suffix(backend + ":*") + platform_devices = backend_for_selector + ":*" for line in get_sycl_ls_verbose(platform_devices, env).stdout.splitlines(): if re.match(r" *Architecture:", line): @@ -1114,6 +1116,13 @@ def get_sycl_ls_verbose(sycl_device, env): features.update(device_family) be, dev = sycl_device.split(":") + if dev.isdigit(): + backend_devices = available_devices.get(be, "gpu") + if isinstance(backend_devices, tuple): + # arch-selection is typically used to select gpu device + dev = "gpu" + else: + dev = backend_devices features.add(dev.replace("fpga", "accelerator")) if "level_zero_v2" in full_name: features.add("level_zero_v2_adapter")