mirror of
https://github.com/huggingface/candle.git
synced 2025-06-22 04:22:50 +00:00
working bfloat matmul
This commit is contained in:
@ -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())
|
||||
}
|
||||
|
Reference in New Issue
Block a user