candle 0.3.0 (#881)

tracel-ai · Oct 20, 2023 · 01d4262 · 01d4262
1 parent af813d0
commit 01d4262
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 6 deletions.
diff --git a/burn-candle/Cargo.toml b/burn-candle/Cargo.toml
@@ -16,8 +16,7 @@ version = "0.10.0"
 derive-new = { workspace = true }
 burn-tensor = { path = "../burn-tensor", version = "0.10.0", default-features = false }
 half = { workspace = true }
-# candle-core = { version = "0.1.2" }
-candle-core = { git = "https://github.com/huggingface/candle", rev = "237323c" }
+candle-core = { version = "0.3.0" }
 
 [dev-dependencies]
 burn-autodiff = { path = "../burn-autodiff", version = "0.10.0", default-features = false, features = [

diff --git a/burn-candle/src/lib.rs b/burn-candle/src/lib.rs
@@ -59,7 +59,7 @@ mod tests {
     burn_tensor::testgen_clamp!();
     burn_tensor::testgen_cos!();
     // burn_tensor::testgen_div!();
-    // burn_tensor::testgen_erf!();
+    burn_tensor::testgen_erf!();
     burn_tensor::testgen_exp!();
     burn_tensor::testgen_flatten!();
     burn_tensor::testgen_full!();
@@ -121,7 +121,7 @@ mod tests {
     burn_autodiff::testgen_ad_cos!();
     burn_autodiff::testgen_ad_cross_entropy_loss!();
     burn_autodiff::testgen_ad_div!();
-    // burn_autodiff::testgen_ad_erf!();
+    burn_autodiff::testgen_ad_erf!();
     burn_autodiff::testgen_ad_exp!();
     // burn_autodiff::testgen_ad_slice!();
     burn_autodiff::testgen_ad_gather_scatter!();

diff --git a/burn-candle/src/ops/tensor.rs b/burn-candle/src/ops/tensor.rs
@@ -393,8 +393,7 @@ impl<F: FloatCandleElement, I: IntCandleElement> TensorOps<CandleBackend<F, I>>
     }
 
     fn erf<const D: usize>(tensor: FloatTensor<Self, D>) -> FloatTensor<Self, D> {
-        // TODO submit an issue at Candle
-        panic!("erf not supported by Candle")
+        CandleTensor::new(tensor.tensor.erf().unwrap())
     }
 
     fn cat<const D: usize>(tensors: Vec<FloatTensor<Self, D>>, dim: usize) -> FloatTensor<Self, D> {