Unverified Commit d3d91b6f authored by Reid's avatar Reid Committed by GitHub
Browse files

[Misc][MacOS] fix bfloat16 error (#18249)


Signed-off-by: default avatarreidliu41 <reid201711@gmail.com>
Co-authored-by: default avatarreidliu41 <reid201711@gmail.com>
parent 87d87147
...@@ -36,7 +36,7 @@ class CpuPlatform(Platform): ...@@ -36,7 +36,7 @@ class CpuPlatform(Platform):
# instead of checking the OS. For instance M2 shall supports bf16 # instead of checking the OS. For instance M2 shall supports bf16
# already. But we need to modify `cpu_extension.cmake` to activate # already. But we need to modify `cpu_extension.cmake` to activate
# the feature in the build. # the feature in the build.
return [torch.bfloat16, torch.float32] return [torch.float16, torch.float32]
# x86/aarch64 CPU has supported both bf16 and fp16 natively. # x86/aarch64 CPU has supported both bf16 and fp16 natively.
return [torch.bfloat16, torch.float16, torch.float32] return [torch.bfloat16, torch.float16, torch.float32]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment