Bump the ug dependency. (#2720)

* Bump the ug dependency. * Fix some test. * Fix the ug test.
huggingface · Jan 16, 2025 · 6fd2f63 · 6fd2f63
1 parent efd0e68
commit 6fd2f63
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -70,9 +70,9 @@ 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"
-ug-metal = "0.0.2"
+ug = "0.1.0"
+ug-cuda = "0.1.0"
+ug-metal = "0.1.0"
 yoke = { version = "0.7.2", features = ["derive"] }
 zip = { version = "1.1.1", default-features = false }
 metal = { version = "0.27.0", features = ["mps"]}

diff --git a/candle-core/tests/custom_op_tests.rs b/candle-core/tests/custom_op_tests.rs
@@ -158,7 +158,7 @@ fn ug_op() -> Result<()> {
         let st = op::store(ptr.id(), layout, src)?;
         let kernel = op::Kernel::new("exp".to_string(), vec![ptr], vec![st]);
         let opts: ug::lower_op::Opts = Default::default();
-        kernel.lower(&opts.with_global(0, 12))?
+        kernel.lower(&opts)?
     };
     let device = if candle_core::utils::cuda_is_available() {
         Device::new_cuda(0)?