From e5bf9d208c6118c1e7a450f744a7fcfcc0e3e696 Mon Sep 17 00:00:00 2001 From: Eunju Yang Date: Tue, 17 Dec 2024 13:46:08 +0900 Subject: [PATCH] [ 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 --- nntrainer/layers/layer_context.h | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/nntrainer/layers/layer_context.h b/nntrainer/layers/layer_context.h index 0ce3b5bc15..fef7ff6d75 100644 --- 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; }