Bump the ug dependency. (#2720)

* Bump the ug dependency.

* Fix some test.

* Fix the ug test.

2 files changed, 4 insertions, 4 deletions

diff --git a/Cargo.toml b/Cargo.toml
index c551d65e..e8d1f769 100644
--- 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
index 3572a4c9..3fc45971 100644
--- 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)?