fix typo for NF

microsoft · May 20, 2024 · 64adc99 · 64adc99
1 parent d1b9dbf
commit 64adc99
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/python/bitblas/ops/general_matmul.py b/python/bitblas/ops/general_matmul.py
@@ -162,16 +162,15 @@ def __initialize_zeros_mode(self, zeros_mode: Optional[str]):
             object.__setattr__(self, "zeros_mode", "original")
 
     def __initialize_fast_decoding(self, fast_decoding: Optional[bool]):
-        if (
+        if fast_decoding is not None:
+            object.__setattr__(self, "fast_decoding", fast_decoding)
+        elif (
             "int" not in self.W_dtype
-            or "nf" not in self.W_dtype
             or self.W_dtype == self.A_dtype
         ):
             object.__setattr__(self, "fast_decoding", False)
         else:
             object.__setattr__(self, "fast_decoding", True)
-        if fast_decoding is not None:
-            object.__setattr__(self, "fast_decoding", fast_decoding)
 
     def __post_init__(self):
         # set M to default dynamic range if it is None