Shuffle the shape bits around.

2025-06-17 19:18:50 +00:00 · 2023-06-19 20:22:12 +01:00
parent 634e0c88ae
commit 01eeb0e72f
5 changed files with 159 additions and 63 deletions
--- a/src/dtype.rs
+++ b/src/dtype.rs
@ -12,3 +12,15 @@ impl DType {
        }
    }
 }
+
+pub trait WithDType {
+    const DTYPE: DType;
+}
+
+impl WithDType for f32 {
+    const DTYPE: DType = DType::F32;
+}
+
+impl WithDType for f64 {
+    const DTYPE: DType = DType::F64;
+}