fix UT bug

intel · May 21, 2024 · 9752312 · 9752312
1 parent 451aa2d
commit 9752312
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/bestla/bestla/bestla_prologue_b.h b/bestla/bestla/bestla_prologue_b.h
@@ -628,6 +628,8 @@ class WeightKBlockNInteger {
     if (qtype == BTLA_DTYPE::S4_CLIP) return compressBit4Weight(N, K, B, dstptr, qtype, threading);
     if (qtype == BTLA_DTYPE::S3_CLIP) return compressBit3Weight(N, K, B, dstptr, qtype, threading);
     if (qtype == BTLA_DTYPE::S2_CLIP) return compressBit2Weight(N, K, B, dstptr, qtype, threading);
+    if (qtype == BTLA_DTYPE::F4_BNB || qtype == BTLA_DTYPE::F4_NF4 || qtype == BTLA_DTYPE::F4_E2M1)
+      return compressBit4Weight(N, K, B, dstptr, qtype, threading);
   }
 
   template <typename RED_T>

diff --git a/bestla/bestla/ut/bestla_prologue_b.cpp b/bestla/bestla/ut/bestla_prologue_b.cpp
@@ -411,7 +411,7 @@ class UT_TransposeBlockQuantize_F4 {
   }
 };
 #ifdef BTLA_UT_PROLOGUE_B
-static UT_TransposeBlockQuantize_F4 sUT_TransposeBlockQuantize_F4;
+static UT_TransposeBlockQuantize_F4 sUT_TransposeBlockQuantize_F4; 
 #endif
 
 class UT_BlockQuantize_INT4 {