File tree Expand file tree Collapse file tree 2 files changed +8
-1
lines changed Expand file tree Collapse file tree 2 files changed +8
-1
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff line change @@ -707,7 +707,7 @@ def remove_level_zero_suffix(devices):
707707available_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 ():
You can’t perform that action at this time.
0 commit comments