This repository was archived by the owner on Oct 25, 2024. It is now read-only.
File tree 2 files changed +7
-9
lines changed
intel_extension_for_transformers/tools
2 files changed +7
-9
lines changed Original file line number Diff line number Diff line change @@ -60,12 +60,10 @@ def get_gpu_family():
60
60
elif 'Arc(TM)' in name :
61
61
result = 'arc'
62
62
else :
63
- assert False , "Unsupported GPU device: {}" .format (name )
63
+ print ("Warning: Unsupported GPU device: {}" .format (name ))
64
+ result = ""
64
65
65
- if result not in supported_gpus ():
66
- assert False , "Unsupported GPU device: {}" .format (name )
67
- else :
68
- return result
66
+ return result
69
67
70
68
_ipex_available = importlib .util .find_spec ("intel_extension_for_pytorch" ) is not None
71
69
_ipex_version = "N/A"
Original file line number Diff line number Diff line change @@ -39,10 +39,10 @@ def get_gpu_family():
39
39
return 'max'
40
40
if 'Arc(TM)' in name :
41
41
return 'arc'
42
- if '0x7d55' in name or '0x7dd5' in name or '0x7d45' in name :
43
- return 'mtl'
44
- assert False , " Unsupported GPU device: {}" .format (name )
45
-
42
+ # if '0x7d55' in name or '0x7dd5' in name or '0x7d45' in name:
43
+ # return 'mtl'
44
+ print ( "Warning: Unsupported GPU device: {}" .format (name ) )
45
+ return ""
46
46
47
47
def check_env_flag (name : str , default : bool = False ) -> bool :
48
48
if default : # if a flag meant to be true if not set / mal-formatted
You can’t perform that action at this time.
0 commit comments