Fix zeros impl

EricLBuehler · Oct 10, 2024 · e51ecb6 · e51ecb6
1 parent 60eb251
commit e51ecb6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/candle-core/src/cuda_backend/device.rs b/candle-core/src/cuda_backend/device.rs
@@ -253,8 +253,8 @@ impl BackendDevice for CudaDevice {
                 CudaStorageSlice::F64(data)
             }
             DType::F8E4M3 => {
-                let data = self.alloc_zeros::<f64>(elem_count).w()?;
-                CudaStorageSlice::F64(data)
+                let data = self.alloc_zeros::<F8E4M3>(elem_count).w()?;
+                CudaStorageSlice::F8E4M3(data)
             }
         };
         Ok(CudaStorage {