Skip to content

Commit 1f92040

Browse files
committed
Git format
1 parent 7ce23a4 commit 1f92040

File tree

1 file changed

+17
-5
lines changed

1 file changed

+17
-5
lines changed

sycl/test-e2e/lit.cfg.py

Lines changed: 17 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -177,15 +177,23 @@
177177
device_name_arch_map = {
178178
# <Device name> : Set of architectures types (and aliases)
179179
# DG1
180-
"gpu-intel-dg1" : {"intel_gpu_dg1"},
180+
"gpu-intel-dg1": {"intel_gpu_dg1"},
181181
# DG2
182-
"gpu-intel-dg2" : {"intel_gpu_acm_g12","intel_gpu_dg2_g12","intel_gpu_acm_g11","intel_gpu_dg2_g11","intel_gpu_acm_g10","intel_gpu_dg2_g10"},
182+
"gpu-intel-dg2": {
183+
"intel_gpu_acm_g12",
184+
"intel_gpu_dg2_g12",
185+
"intel_gpu_acm_g11",
186+
"intel_gpu_dg2_g11",
187+
"intel_gpu_acm_g10",
188+
"intel_gpu_dg2_g10",
189+
},
183190
# Gen12
184-
"gpu-intel-gen12" : {"intel_gpu_tgllp","intel_gpu_tgl"},
191+
"gpu-intel-gen12": {"intel_gpu_tgllp", "intel_gpu_tgl"},
185192
# Gen11
186-
"gpu-intel-gen11" : {"intel_gpu_icllp","intel_gpu_icl"},
193+
"gpu-intel-gen11": {"intel_gpu_icllp", "intel_gpu_icl"},
187194
}
188195

196+
189197
def get_device_name_from_arch(arch):
190198
for device_name, arch_set in device_name_arch_map.items():
191199
if arch in arch_set:
@@ -789,7 +797,11 @@ def open_check_file(file_name):
789797
architecture_feature = set("arch-" + s for s in architectures)
790798
# Add device name features like intel-gpu-gen12, intel-gpu-dg2 based on
791799
# the architecture reported by sycl-ls.
792-
device_names = set(get_device_name_from_arch(arch) for arch in architectures if get_device_name_from_arch(arch) is not None)
800+
device_names = set(
801+
get_device_name_from_arch(arch)
802+
for arch in architectures
803+
if get_device_name_from_arch(arch) is not None
804+
)
793805
features = set()
794806
features.update(aspect_features)
795807
features.update(sg_size_features)

0 commit comments

Comments
 (0)