diff options
| author | Ibiyemi Abiodun <ibiyemi@intulon.com> | 2023-10-24 12:06:03 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-10-24 17:06:03 +0100 |
| commit | 45dbe541bcde2fa12281a08df501baf76b11558f (patch) | |
| tree | 0d5c7fb06e24a49e1dc5b0be4a62faf84886aae1 | |
| parent | 7bd0faba7592a150f2c29c1f754eb27529d1d6d1 (diff) | |
| download | candle-45dbe541bcde2fa12281a08df501baf76b11558f.tar.gz candle-45dbe541bcde2fa12281a08df501baf76b11558f.tar.bz2 candle-45dbe541bcde2fa12281a08df501baf76b11558f.zip | |
fix ucopy for `f64` tensors (#1170)
| -rw-r--r-- | candle-core/src/cuda_backend.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-core/src/cuda_backend.rs b/candle-core/src/cuda_backend.rs index f1c35ae1..41ddaed3 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. |