Only optimize float tensors. (#1069)

2025-06-16 10:38:54 +00:00 · 2023-10-10 10:05:41 +02:00
parent bc3351bce4
commit 9fea56d28e
2 changed files with 19 additions and 0 deletions
--- a/candle-core/src/dtype.rs
+++ b/candle-core/src/dtype.rs
@ -67,6 +67,20 @@ impl DType {
            Self::F64 => 8,
        }
    }
+
+    pub fn is_int(&self) -> bool {
+        match self {
+            Self::U8 | Self::U32 | Self::I64 => true,
+            Self::BF16 | Self::F16 | Self::F32 | Self::F64 => false,
+        }
+    }
+
+    pub fn is_float(&self) -> bool {
+        match self {
+            Self::U8 | Self::U32 | Self::I64 => false,
+            Self::BF16 | Self::F16 | Self::F32 | Self::F64 => true,
+        }
+    }
 }

 pub trait WithDType: