From 45dbe541bcde2fa12281a08df501baf76b11558f Mon Sep 17 00:00:00 2001
From: Ibiyemi Abiodun <ibiyemi@intulon.com>
Date: Tue, 24 Oct 2023 12:06:03 -0400
Subject: [PATCH] fix ucopy for `f64` tensors (#1170)

---
 candle-core/src/cuda_backend.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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.