huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use the patched gemm for the time being.

Browse Source
This commit is contained in:
laurent
2023-07-03 10:29:15 +01:00
parent ec4871b8a4
commit 639270b796
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-core/Cargo.toml
View File

@ -14,7 +14,9 @@ readme = "README.md"
byteorder = "1.4.3"
candle-kernels = { path = "../candle-kernels", optional = true }
cudarc = { version = "0.9.9", optional = true, features = ["f16"] }
gemm = "0.15.4"
# TODO: Switch back to the official gemm implementation once something similar to
# https://github.com/sarah-ek/gemm/pull/8 is available.
gemm = { git = "https://github.com/LaurentMazare/gemm.git", branch = "f16-vectorize-pack" }
half = { version = "2.3.1", features = ["num-traits"] }
memmap2 = "0.7.1"
num-traits = "0.2.15"