Revert "Bump the ug dependency. (huggingface#2720)"

This reverts commit 6fd2f63.

Author: sgrebnov

Cargo.toml

@@ -70,9 +70,9 @@ tokenizers = { version = "0.21.0", default-features = false }
 tracing = "0.1.37"
 tracing-chrome = "0.7.1"
 tracing-subscriber = "0.3.7"
-ug = "0.1.0"
-ug-cuda = "0.1.0"
-ug-metal = "0.1.0"
+ug = "0.0.2"
+ug-cuda = "0.0.2"
+ug-metal = "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"]}

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)?
+        kernel.lower(&opts.with_global(0, 12))?
     };
     let device = if candle_core::utils::cuda_is_available() {
         Device::new_cuda(0)?