summaryrefslogtreecommitdiff
path: root/candle-metal-kernels/Cargo.toml
diff options
context:
space:
mode:
authorLaurent Mazare <laurent.mazare@gmail.com>2024-09-11 21:52:37 +0100
committerGitHub <noreply@github.com>2024-09-11 22:52:37 +0200
commit0cb0bd1dfaac97425a0805a2e7f24ea7992236c0 (patch)
tree4bd20985fa2333efa01dc6ade4789812283979bf /candle-metal-kernels/Cargo.toml
parentafb6575835599938248c027f50a8100c289a1a96 (diff)
downloadcandle-0cb0bd1dfaac97425a0805a2e7f24ea7992236c0.tar.gz
candle-0cb0bd1dfaac97425a0805a2e7f24ea7992236c0.tar.bz2
candle-0cb0bd1dfaac97425a0805a2e7f24ea7992236c0.zip
Add some metal gemm benchark. (#2471)
* Add some metal gemm benchark. * More benchmarks.
Diffstat (limited to 'candle-metal-kernels/Cargo.toml')
-rw-r--r--candle-metal-kernels/Cargo.toml2
1 files changed, 2 insertions, 0 deletions
diff --git a/candle-metal-kernels/Cargo.toml b/candle-metal-kernels/Cargo.toml
index 772452c9..8f92099d 100644
--- a/candle-metal-kernels/Cargo.toml
+++ b/candle-metal-kernels/Cargo.toml
@@ -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"