fix

Author: metascroy

kernels/quantized/cpu/embeddingxb.cpp

@@ -6,6 +6,7 @@
  * LICENSE file in the root directory of this source tree.
  */
 
+#include <executorch/kernels/quantized/cpu/embeddingxb.h>
 #include <executorch/runtime/kernel/kernel_includes.h>
 #include <algorithm>
 #include <cassert>

kernels/quantized/test/op_embedding2b_test.cpp

@@ -169,14 +169,13 @@ TEST(OpQuantizedEmbedding2bTest, TestGroupWiseQuantizedEmbeddingDeath3) {
   int64_t quant_min = -2;
   int64_t quant_max = 1;
 
-  Tensor weight_scales =
-      tf.make({2, 3}, {1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0});
-  Tensor weight_zero_points = tf.make({2, 3}, {0, 0, 0, 0, 0, 0, 0, 0, 0});
+  Tensor weight_scales = tf.make({2, 3}, {1.0, 1.0, 1.0, 1.0, 1.0, 1.0});
+  Tensor weight_zero_points = tf.make({2, 3}, {0, 0, 0, 0, 0, 0});
   Tensor qweight = tfb.make({2, 1}, {236, 134, 228});
   Tensor indices = tfl.make({2}, {0, 2});
   Tensor out = tf.zeros({2, 8});
 
-  // scales/zeros imply groupsize 3, which does not divide embed dimension from
+  // scales/zeros imply 3 groups, which does not divide embed dimension from
   // qvals (8)
   ET_EXPECT_DEATH(
       quantized_embedding_2bit_out(