Residual fixes

Author: EricLBuehler

candle-core/src/cuda_backend/mod.rs

@@ -1349,7 +1349,7 @@ impl BackendStorage for CudaStorage {
             S::F16(s) => (slice_ptr(s, src_o), "const_set_f16"),
             S::F32(s) => (slice_ptr(s, src_o), "const_set_f32"),
             S::F64(s) => (slice_ptr(s, src_o), "const_set_f64"),
-            S::F8E4M3(s) => (slice_ptr(s, src_o), "const_set_f8e4m3"),
+            S::F8E4M3(s) => (slice_ptr(s, src_o), "const_set_f8_e4m3"),
             S::F4(_) | S::F6E2M3(_) | S::F6E3M2(_) | S::F8E8M0(_) => {
                 return Err(CudaError::UnsupportedDtype {
                     dtype: self.dtype(),

candle-core/src/safetensors.rs

@@ -630,15 +630,15 @@ mod tests {
     }
 
     #[test]
-    fn load_i8() {
-        let bytes = b"8\0\0\0\0\0\0\0{\"x\":{\"dtype\":\"I8\",\"shape\":[2],\"data_offsets\":[0,2]}}   \x01\x03";
-        std::fs::write("test_i8.safetensors", bytes).unwrap();
-        let weights = load("test_i8.safetensors", &Device::Cpu).unwrap();
+    fn load_u8() {
+        let bytes = b"8\0\0\0\0\0\0\0{\"x\":{\"dtype\":\"U8\",\"shape\":[2],\"data_offsets\":[0,2]}}   \x01\x03";
+        std::fs::write("test_u8.safetensors", bytes).unwrap();
+        let weights = load("test_u8.safetensors", &Device::Cpu).unwrap();
         let tensor = weights.get("x").unwrap();
         assert_eq!(tensor.dims(), &[2]);
-        assert_eq!(tensor.dtype(), DType::I64);
-        let data: Vec<i64> = tensor.to_vec1().unwrap();
+        assert_eq!(tensor.dtype(), DType::U8);
+        let data: Vec<u8> = tensor.to_vec1().unwrap();
         assert_eq!(data, vec![1, 3]);
-        std::fs::remove_file("test_i8.safetensors").unwrap();
+        std::fs::remove_file("test_u8.safetensors").unwrap();
     }
 }