From deb60910993f756865618da68865b9e26b8f826a Mon Sep 17 00:00:00 2001
From: laurent <laurent.mazare@gmail.com>
Date: Wed, 21 Jun 2023 19:50:00 +0100
Subject: [PATCH] Use a type alias for cuda errors.

---
 src/cuda_backend.rs | 2 ++
 src/error.rs        | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/cuda_backend.rs b/src/cuda_backend.rs
index 44e75575..5a80df08 100644
--- a/src/cuda_backend.rs
+++ b/src/cuda_backend.rs
@@ -1,6 +1,8 @@
 use crate::{CpuStorage, DType, Result, Shape};
 use cudarc::driver::CudaSlice;
 
+pub(crate) type Error = cudarc::driver::DriverError;
+
 #[derive(Debug, Clone)]
 pub struct CudaDevice(std::sync::Arc<cudarc::driver::CudaDevice>);
 
diff --git a/src/error.rs b/src/error.rs
index 1990b67a..94f5e50d 100644
--- a/src/error.rs
+++ b/src/error.rs
@@ -35,7 +35,7 @@ pub enum Error {
     },
 
     #[error(transparent)]
-    Cudarc(#[from] cudarc::driver::DriverError),
+    Cudarc(#[from] crate::cuda_backend::Error),
 }
 
 pub type Result<T> = std::result::Result<T, Error>;