From 7ba3a8b6fc94394c60a35fbb6ebfb89b2e781e4a Mon Sep 17 00:00:00 2001 From: Corey Lowman Date: Wed, 25 Oct 2023 13:32:57 -0400 Subject: [PATCH] #870 allocate 1 byte for workspace instead of 0 (#876) --- dfdx-core/src/tensor/cuda/device.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dfdx-core/src/tensor/cuda/device.rs b/dfdx-core/src/tensor/cuda/device.rs index 156bdddd..de6f7196 100644 --- 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::(0)?)); + let workspace = Arc::new(Mutex::new(dev.alloc_zeros::(1)?)); Ok(Self { cpu, dev,