build: update cuda-driver

Signed-off-by: YdrMaster <[email protected]>

Author: YdrMaster

Cargo.toml

@@ -12,8 +12,8 @@ infini-op = { git = "https://github.com/InfiniTensor/infini-toolkit", rev = "e83
 infini-ccl = { git = "https://github.com/InfiniTensor/infini-toolkit", rev = "e8362c3" }
 search-infini-tools = { git = "https://github.com/InfiniTensor/infini-toolkit", rev = "e8362c3" }
 
-cuda = { git = "https://github.com/YdrMaster/cuda-driver", rev = "1751f0a" }
-cublas = { git = "https://github.com/YdrMaster/cuda-driver", rev = "1751f0a" }
-nccl = { git = "https://github.com/YdrMaster/cuda-driver", rev = "1751f0a" }
-search-cuda-tools = { git = "https://github.com/YdrMaster/cuda-driver", rev = "1751f0a" }
-search-corex-tools = { git = "https://github.com/YdrMaster/cuda-driver", rev = "1751f0a" }
+cuda = { git = "https://github.com/YdrMaster/cuda-driver", rev = "a815e41" }
+cublas = { git = "https://github.com/YdrMaster/cuda-driver", rev = "a815e41" }
+nccl = { git = "https://github.com/YdrMaster/cuda-driver", rev = "a815e41" }
+search-cuda-tools = { git = "https://github.com/YdrMaster/cuda-driver", rev = "a815e41" }
+search-corex-tools = { git = "https://github.com/YdrMaster/cuda-driver", rev = "a815e41" }

operators/src/rms_norm/common_cpu/mod.rs

@@ -97,15 +97,15 @@ unsafe impl<W, A> Sync for Scheme<W, A> {}
 impl<W, A> Scheme<W, A> {
     #[inline]
     unsafe fn y_ptr(&self, i: isize, j: isize) -> *mut A {
-        self.y.byte_offset(i * self.nsy + j * self.dsy)
+        unsafe { self.y.byte_offset(i * self.nsy + j * self.dsy) }
     }
     #[inline]
     unsafe fn x_ptr(&self, i: isize, j: isize) -> *const A {
-        self.x.byte_offset(i * self.nsx + j * self.dsx)
+        unsafe { self.x.byte_offset(i * self.nsx + j * self.dsx) }
     }
     #[inline]
     unsafe fn w_ptr(&self, j: isize) -> *const W {
-        self.w.byte_offset(j * self.dsw)
+        unsafe { self.w.byte_offset(j * self.dsw) }
     }
 }
 
@@ -125,54 +125,54 @@ impl<W> Scheme<W, f16> {
 
     #[inline]
     unsafe fn y(&self, i: isize, j: isize, val: f32) {
-        self.y_ptr(i, j).write(f16::from_f32(val))
+        unsafe { self.y_ptr(i, j).write(f16::from_f32(val)) }
     }
     #[inline]
     unsafe fn x(&self, i: isize, j: isize) -> f32 {
-        self.x_ptr(i, j).read().to_f32()
+        unsafe { self.x_ptr(i, j).read().to_f32() }
     }
 }
 impl<W> Scheme<W, f32> {
     impl_k!(f32);
 
     #[inline]
     unsafe fn y(&self, i: isize, j: isize, val: f32) {
-        self.y_ptr(i, j).write(val)
+        unsafe { self.y_ptr(i, j).write(val) }
     }
     #[inline]
     unsafe fn x(&self, i: isize, j: isize) -> f32 {
-        self.x_ptr(i, j).read()
+        unsafe { self.x_ptr(i, j).read() }
     }
 }
 impl<W> Scheme<W, f64> {
     impl_k!(f64);
 
     #[inline]
     unsafe fn y(&self, i: isize, j: isize, val: f64) {
-        self.y_ptr(i, j).write(val)
+        unsafe { self.y_ptr(i, j).write(val) }
     }
     #[inline]
     unsafe fn x(&self, i: isize, j: isize) -> f64 {
-        self.x_ptr(i, j).read()
+        unsafe { self.x_ptr(i, j).read() }
     }
 }
 
 impl<A> Scheme<f16, A> {
     #[inline]
     unsafe fn w(&self, j: isize) -> f32 {
-        self.w_ptr(j).read().to_f32()
+        unsafe { self.w_ptr(j).read() }.to_f32()
     }
 }
 impl<A> Scheme<f32, A> {
     #[inline]
     unsafe fn w(&self, j: isize) -> f32 {
-        self.w_ptr(j).read()
+        unsafe { self.w_ptr(j).read() }
     }
 }
 impl<A> Scheme<f64, A> {
     #[inline]
     unsafe fn w(&self, j: isize) -> f64 {
-        self.w_ptr(j).read()
+        unsafe { self.w_ptr(j).read() }
     }
 }