huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-20 20:09:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Prebuild all our kernels.

Browse Source
This commit is contained in:
Nicolas Patry
2024-03-18 16:39:38 +01:00
parent 41416d2376
commit 5ac3302fac
5 changed files with 70 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
candle-core/src/metal_backend.rs
View File

@ -588,6 +588,7 @@ impl BackendStorage for MetalStorage {
(DType::U32, DType::F32) => "cast_u32_f32",
(DType::U32, DType::U8) => "cast_u32_u8",
(DType::U32, DType::I64) => "cast_u32_i64",
(DType::U32, DType::F16) => "cast_u32_f16",
(DType::U32, DType::BF16) => "cast_u32_bf16",
(DType::U8, DType::U32) => "cast_u8_u32",