[ BCQ ] Add a feature to enable BCQ tensor as a weight type (2)

- This patch adds a missing part of #2820. - In order to use BCQ as a weight, this patch should be included. Self-evaluation: Build test: [X]Passed [ ]Failed [ ]Skipped Run test: [X]Passed [ ]Failed [ ]Skipped Signed-off-by: Eunju Yang <[email protected]>
nnstreamer · Dec 19, 2024 · e5bf9d2 · e5bf9d2
1 parent bc8217c
commit e5bf9d2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/nntrainer/layers/layer_context.h b/nntrainer/layers/layer_context.h
@@ -495,7 +495,8 @@ class RunLayerContext {
     Tensor &t_w = weights[idx]->getVariableRef();
 
     if (t_w.getDataType() == Tdatatype::FP32 ||
-        t_w.getDataType() == Tdatatype::FP16) {
+        t_w.getDataType() == Tdatatype::FP16 ||
+        t_w.getDataType() == Tdatatype::BCQ) {
       w = t_w;
       return;
     }