diff options
Diffstat (limited to 'candle-core/src/cuda_backend/mod.rs')
| -rw-r--r-- | candle-core/src/cuda_backend/mod.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-core/src/cuda_backend/mod.rs b/candle-core/src/cuda_backend/mod.rs index 88f325f4..9e72dcc8 100644 --- a/candle-core/src/cuda_backend/mod.rs +++ b/candle-core/src/cuda_backend/mod.rs @@ -16,7 +16,7 @@ mod error; mod utils; pub use device::{CudaDevice, DeviceId}; pub use error::{CudaError, WrapErr}; -pub use utils::{Map1, Map1Any, Map2, Map2Any, Map2InPlace, S}; +pub use utils::{Map1, Map1Any, Map2, Map2Any, Map2InPlace, Map3, S}; pub enum SlicePtrOrNull<T> { Ptr(CudaSlice<T>), |