File tree Expand file tree Collapse file tree 1 file changed +8
-6
lines changed Expand file tree Collapse file tree 1 file changed +8
-6
lines changed Original file line number Diff line number Diff line change @@ -76,12 +76,14 @@ def supported_dtypes(self) -> list[torch.dtype]:
76
76
if self .get_cpu_architecture () == CpuArchEnum .POWERPC :
77
77
return [torch .bfloat16 , torch .float32 ]
78
78
elif (self .get_cpu_architecture () == CpuArchEnum .ARM
79
- and sys .platform .startswith ("darwin" )):
80
- if (subprocess .check_output (
81
- ["sysctl" , "-n" , "-e" , "hw.optional.arm.FEAT_BF16" ]
82
- ).strip () == b"1" ):
83
- return [torch .bfloat16 , torch .float16 , torch .float32 ]
84
- return [torch .float16 , torch .float32 ]
79
+ and sys .platform .startswith ("darwin" )):
80
+ try :
81
+ if (subprocess .check_output (
82
+ ["sysctl" , "-n" , "-i" ,
83
+ "hw.optional.arm.FEAT_BF16" ]).strip () == b"1" ):
84
+ return [torch .bfloat16 , torch .float16 , torch .float32 ]
85
+ except Exception :
86
+ return [torch .float16 , torch .float32 ]
85
87
# x86/aarch64 CPU has supported both bf16 and fp16 natively.
86
88
return [torch .bfloat16 , torch .float16 , torch .float32 ]
87
89
You can’t perform that action at this time.
0 commit comments