diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-08-23 12:58:55 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-08-23 12:58:55 +0100 |
| commit | aba1e90797e430f28eec13b14b76dd5355876f9c (patch) | |
| tree | 16bcf7fb151715d3bcdbec2b5263922bd0bdd35a /candle-flash-attn | |
| parent | 4ee1cf038ada55ec477dcd6496cf2aec1902775b (diff) | |
| download | candle-aba1e90797e430f28eec13b14b76dd5355876f9c.tar.gz candle-aba1e90797e430f28eec13b14b76dd5355876f9c.tar.bz2 candle-aba1e90797e430f28eec13b14b76dd5355876f9c.zip | |
Add some group parameter to convolutions. (#566)
* Add some group parameter to convolutions.
* Avoid some unnecessary groups checks.
* Move the tensor convolution bits.
* Properh handling of groups.
* Bump the crate version.
* And add a changelog.
Diffstat (limited to 'candle-flash-attn')
| -rw-r--r-- | candle-flash-attn/Cargo.toml | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/candle-flash-attn/Cargo.toml b/candle-flash-attn/Cargo.toml index f88a88d5..b0efaf52 100644 --- a/candle-flash-attn/Cargo.toml +++ b/candle-flash-attn/Cargo.toml @@ -1,6 +1,6 @@ [package] name = "candle-flash-attn" -version = "0.1.2" +version = "0.1.3" edition = "2021" description = "Flash attention layer for the candle ML framework." @@ -11,7 +11,7 @@ license = "MIT OR Apache-2.0" readme = "README.md" [dependencies] -candle = { path = "../candle-core", features = ["cuda"], version = "0.1.2", package = "candle-core" } +candle = { path = "../candle-core", features = ["cuda"], version = "0.1.3", package = "candle-core" } half = { version = "2.3.1", features = ["num-traits"] } [build-dependencies] @@ -21,4 +21,4 @@ rayon = "1.7.0" [dev-dependencies] anyhow = { version = "1", features = ["backtrace"] } -candle-nn = { path = "../candle-nn", version = "0.1.2", features = ["cuda"] } +candle-nn = { path = "../candle-nn", version = "0.1.3", features = ["cuda"] } |