diff --git a/candle-core/src/cuda_backend.rs b/candle-core/src/cuda_backend.rs index f1c35ae170..41ddaed31d 100644 --- a/candle-core/src/cuda_backend.rs +++ b/candle-core/src/cuda_backend.rs @@ -2171,7 +2171,7 @@ impl BackendStorage for CudaStorage { if src_l.is_contiguous() { dev.dtod_copy(&src, &mut dst).w()? } else { - let func = dev.get_or_load_func("ucopy_64", kernels::UNARY)?; + let func = dev.get_or_load_func("ucopy_f64", kernels::UNARY)?; // SAFETY: Set later by running the kernel. let params = (el_count, dims.len(), &ds, &src, &mut dst); // SAFETY: ffi.