#870 allocate 1 byte for workspace instead of 0 (#876)

coreylowman · Oct 25, 2023 · 7ba3a8b · 7ba3a8b
1 parent 7df12c0
commit 7ba3a8b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dfdx-core/src/tensor/cuda/device.rs b/dfdx-core/src/tensor/cuda/device.rs
@@ -76,7 +76,7 @@ impl Cuda {
         #[cfg(feature = "cudnn")]
         let cudnn = cudarc::cudnn::Cudnn::new(dev.clone())?;
         let par_stream = Arc::new(dev.fork_default_stream()?);
-        let workspace = Arc::new(Mutex::new(dev.alloc_zeros::<u8>(0)?));
+        let workspace = Arc::new(Mutex::new(dev.alloc_zeros::<u8>(1)?));
         Ok(Self {
             cpu,
             dev,