From 594d984f9cf79207f3beb6114ddf73cbc8427b56 Mon Sep 17 00:00:00 2001
From: Laurent Mazare <laurent.mazare@gmail.com>
Date: Sun, 27 Oct 2024 13:37:19 +0100
Subject: Support for UG kernels. (#2579)

* Support for UG kernels.

* Add a dedicated test.
---
 Cargo.toml | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'Cargo.toml')

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"]}
-- 
cgit v1.2.3