working bfloat matmul

This commit is contained in:
Ivar Flakstad
2024-03-18 14:38:40 +01:00
parent 0c09d10f32
commit c974dee369
6 changed files with 32 additions and 16 deletions

View File

@ -1254,6 +1254,7 @@ impl BackendStorage for MetalStorage {
let name = match self.dtype {
DType::F32 => "sgemm",
DType::F16 => "hgemm",
DType::BF16 => "bgemm",
dtype => {
return Err(MetalError::Message(format!("matmul doesn't support {dtype:?}")).into())
}