diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-07-03 10:36:09 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-07-03 10:36:09 +0100 |
| commit | 48089005f6585c2f603f0d930a4a94b5aa18402f (patch) | |
| tree | a16495cd9f248753aecfea22ac159cd22f046868 | |
| parent | ec4871b8a41615546ea8b42638c56ce64a9fdf72 (diff) | |
| parent | 639270b7966f550abf9d4916fff17dd6e8cfd306 (diff) | |
| download | candle-48089005f6585c2f603f0d930a4a94b5aa18402f.tar.gz candle-48089005f6585c2f603f0d930a4a94b5aa18402f.tar.bz2 candle-48089005f6585c2f603f0d930a4a94b5aa18402f.zip | |
Merge pull request #58 from LaurentMazare/use-patched-gemm
Use the patched gemm for the time being.
| -rw-r--r-- | candle-core/Cargo.toml | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/candle-core/Cargo.toml b/candle-core/Cargo.toml index 7076e4e4..ca3e75f1 100644 --- a/candle-core/Cargo.toml +++ b/candle-core/Cargo.toml @@ -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" |