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

Add some metal gemm benchark. (#2471)

Browse Source 
* Add some metal gemm benchark.

* More benchmarks.
This commit is contained in:
Laurent Mazare
2024-09-11 21:52:37 +01:00
committed by GitHub
parent afb6575835
commit 0cb0bd1dfa
2 changed files with 138 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -17,10 +17,12 @@ thiserror = "1"
tracing = "0.1.37"
[dev-dependencies]
clap = { version = "4.2.4", features = ["derive"] }
half = { version = "2.3.1", features = [
"num-traits",
"use-intrinsics",
"rand_distr",
] }
anyhow = "1"
rand = "0.8.5"
rand_distr = "0.4.3"