diff options
Diffstat (limited to 'candle-core/examples/cuda_sum_benchmark.rs')
| -rw-r--r-- | candle-core/examples/cuda_sum_benchmark.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-core/examples/cuda_sum_benchmark.rs b/candle-core/examples/cuda_sum_benchmark.rs index bb4dc8be..1c8b0136 100644 --- a/candle-core/examples/cuda_sum_benchmark.rs +++ b/candle-core/examples/cuda_sum_benchmark.rs @@ -4,7 +4,7 @@ extern crate intel_mkl_src; use std::str::FromStr; use anyhow::Result; -use candle::{Device, Tensor}; +use candle_core::{Device, Tensor}; fn cos_sin(n: usize, device: &Device) -> Result<Tensor> { let thetas: Vec<_> = (0..n).map(|i| (i as f32 / n as f32)).collect(); |