summaryrefslogtreecommitdiff
path: root/Cargo.toml
diff options
context:
space:
mode:
authorLaurent Mazare <laurent.mazare@gmail.com>2024-10-27 13:37:19 +0100
committerGitHub <noreply@github.com>2024-10-27 13:37:19 +0100
commit594d984f9cf79207f3beb6114ddf73cbc8427b56 (patch)
treed82d510f8a253dd340fd83ac27cf190d606fd44b /Cargo.toml
parent37e0ab8c64eb8219e32cf546ac2aa570ed3d1f82 (diff)
downloadcandle-594d984f9cf79207f3beb6114ddf73cbc8427b56.tar.gz
candle-594d984f9cf79207f3beb6114ddf73cbc8427b56.tar.bz2
candle-594d984f9cf79207f3beb6114ddf73cbc8427b56.zip
Support for UG kernels. (#2579)
* Support for UG kernels. * Add a dedicated test.
Diffstat (limited to 'Cargo.toml')
-rw-r--r--Cargo.toml2
1 files changed, 2 insertions, 0 deletions
diff --git a/Cargo.toml b/Cargo.toml
index bd6e1a85..64e1460e 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -70,6 +70,8 @@ tokenizers = { version = "0.19.1", default-features = false }
tracing = "0.1.37"
tracing-chrome = "0.7.1"
tracing-subscriber = "0.3.7"
+ug = "0.0.2"
+ug-cuda = "0.0.2"
yoke = { version = "0.7.2", features = ["derive"] }
zip = { version = "1.1.1", default-features = false }
metal = { version = "0.27.0", features = ["mps"]}