huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 18:48:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix ucopy for f64 tensors (#1170)

Browse Source
This commit is contained in:
Ibiyemi Abiodun
2023-10-24 12:06:03 -04:00
committed by GitHub
parent 7bd0faba75
commit 45dbe541bc
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-core/src/cuda_backend.rs
View File

@ -2171,7 +2171,7 @@ impl BackendStorage for CudaStorage {
if src_l.is_contiguous() { if src_l.is_contiguous() {
dev.dtod_copy(&src, &mut dst).w()? dev.dtod_copy(&src, &mut dst).w()?
} else { } 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. // SAFETY: Set later by running the kernel.
let params = (el_count, dims.len(), &ds, &src, &mut dst); let params = (el_count, dims.len(), &ds, &src, &mut dst);
// SAFETY: ffi. // SAFETY: ffi.