Add correct int type to the quantized weights matrix in MatMulInteger (…

…kraiskil#29) * Add correct int type to the quantized weights matrix in MatMulInteger
GorrG · Feb 13, 2023 · bdfe0f7 · bdfe0f7
1 parent 87a5480
commit bdfe0f7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/nodes/matmulinteger.h b/src/nodes/matmulinteger.h
@@ -49,6 +49,7 @@ class MatMulInteger : public Node {
 	{
 		std::string intype = A->data_type_str();
 		std::string outtype = Y->data_type_str();
+		std::string weighttype = B->data_type_str();
 		std::string a_zero;
 		std::string b_zero;
 
@@ -78,7 +79,7 @@ class MatMulInteger : public Node {
 
 		INDT_1 "/*MatMulInteger*/" << std::endl;
 		INDT_1 << intype << " *A = (" << intype << "*)" << A->cname() << ";" << std::endl;
-		INDT_1 << intype << " *B = (" << intype << "*)" << B->cname() << ";" << std::endl;
+		INDT_1 << weighttype << " *B = (" << weighttype << "*)" << B->cname() << ";" << std::endl;
 		INDT_1 << outtype << " *Y = (" << outtype << "*)" << Y->cname() << ";" << std::endl;
 
 		INDT_1 << "for( uint32_t r=0; r<" << rows << "; r++ )" << std::endl;