diff options
Diffstat (limited to 'tests/grad_tests.rs')
| -rw-r--r-- | tests/grad_tests.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/grad_tests.rs b/tests/grad_tests.rs index 432b1520..56186e5d 100644 --- a/tests/grad_tests.rs +++ b/tests/grad_tests.rs @@ -3,7 +3,7 @@ use candle::{Device, Tensor}; #[test] fn simple_grad() -> Result<()> { - let x = Tensor::var(&[3f32, 1., 4.], Device::Cpu)?; + let x = Tensor::var(&[3f32, 1., 4.], &Device::Cpu)?; let y = (((&x * &x)? + &x * 5f64)? + 4f64)?; let grads = y.backward()?; let grad_x = grads.get(&x).context("no grad for x")?; |