huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-15 18:28:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(bf16): add cast support + tests for cast + bin ops (#1524)

Browse Source
This commit is contained in:
Kyle McCarthy
2024-01-11 08:49:13 -06:00
committed by GitHub
parent 9f0c99f0c1
commit 402349d120
6 changed files with 243 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
candle-metal-kernels/Cargo.toml
View File

@ -9,12 +9,17 @@ keywords = ["blas", "tensor", "machine-learning"]
categories = ["science"]
license = "MIT OR Apache-2.0"
[dependencies]
metal = { version = "0.27.0", features = ["mps"]}
metal = { version = "0.27.0", features = ["mps"] }
once_cell = "1.18.0"
thiserror = "1"
tracing = "0.1.37"
[dev-dependencies]
half = { version = "2.3.1", features = ["num-traits", "use-intrinsics", "rand_distr"] }
half = { version = "2.3.1", features = [
"num-traits",
"use-intrinsics",
"rand_distr",
] }
rand = "0.8.5"