File tree Expand file tree Collapse file tree 1 file changed +7
-1
lines changed Expand file tree Collapse file tree 1 file changed +7
-1
lines changed Original file line number Diff line number Diff line change @@ -719,9 +719,15 @@ def open_check_file(file_name):
719719 sg_sizes_str = line .strip ().replace ("info::device::sub_group_sizes: " , "" )
720720 dev_sg_sizes .append (sg_sizes_str .strip ().split (" " ))
721721 if re .match (r" *DeviceID*" , line ):
722+ gpu_intel_pvc_1T_device_id = "3034"
723+ gpu_intel_pvc_2T_device_id = "3029"
724+
722725 _ , device_id = line .strip ().split (":" , 1 )
723- if device_id .strip () == "3034" :
726+ if device_id .strip () == gpu_intel_pvc_1T_device_id :
724727 config .available_features .add ("gpu-intel-pvc-1T" )
728+ if device_id .strip () == gpu_intel_pvc_2T_device_id :
729+ config .available_features .add ("gpu-intel-pvc-2T" )
730+
725731 if re .match (r" *Architecture:" , line ):
726732 _ , architecture = line .strip ().split (":" , 1 )
727733 architectures .add (architecture .strip ())
You can’t perform that action at this time.
0 commit comments