Revert the bf16 gemm metal changes for now. (#2386)

This commit is contained in:
Laurent Mazare
2024-08-01 22:08:47 +01:00
committed by GitHub
parent 6991a37b94
commit 0fcb40b229
3 changed files with 23 additions and 21 deletions

View File

@ -173,8 +173,8 @@ impl Device {
pub fn supports_bf16(&self) -> bool {
match self {
Self::Cuda(_) | Self::Metal(_) => true,
Self::Cpu => false,
Self::Cuda(_) => true,
Self::Metal(_) | Self::Cpu => false,
}
}