From 0ae900a954ee19ebb8d33ba57493e4a14249c719 Mon Sep 17 00:00:00 2001 From: Kenneth Heafield Date: Sat, 23 Jun 2018 20:27:52 +0100 Subject: Name Generic_16bit to Int16, Generic_8bit to Int8 --- example.cc | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) (limited to 'example.cc') diff --git a/example.cc b/example.cc index c67e48b..0072596 100644 --- a/example.cc +++ b/example.cc @@ -41,14 +41,14 @@ int main() { AlignedVector A_prepared(A_rows * width); AlignedVector B_prepared(width * B_cols); // Quantize A. - intgemm::Generic_16bit::PrepareA(A.get(), A_prepared.get(), quant_mult, A_rows, width); + intgemm::Int16::PrepareA(A.get(), A_prepared.get(), quant_mult, A_rows, width); // Quantize and reshape B. // Typically you will do this once when parameters are loaded, not every time. - intgemm::Generic_16bit::PrepareB(B.get(), B_prepared.get(), quant_mult, width, B_cols); + intgemm::Int16::PrepareB(B.get(), B_prepared.get(), quant_mult, width, B_cols); AlignedVector C(A_rows * B_cols); // Do the actual multiply. - intgemm::Generic_16bit::Multiply(A_prepared.get(), B_prepared.get(), C.get(), 1.0 / (quant_mult * quant_mult), A_rows, width, B_cols); + intgemm::Int16::Multiply(A_prepared.get(), B_prepared.get(), C.get(), 1.0 / (quant_mult * quant_mult), A_rows, width, B_cols); // Sanity check. C will be row major. assert(fabs(C[0] - top_left_reference) < 0.05); } @@ -60,14 +60,14 @@ int main() { AlignedVector A_prepared(A_rows * width); AlignedVector B_prepared(width * B_cols); // Quantize A. - intgemm::Generic_8bit::PrepareA(A.get(), A_prepared.get(), quant_mult, A_rows, width); + intgemm::Int8::PrepareA(A.get(), A_prepared.get(), quant_mult, A_rows, width); // Quantize and reshape B. // Typically you will do this once when parameters are loaded, not every time. - intgemm::Generic_8bit::PrepareB(B.get(), B_prepared.get(), quant_mult, width, B_cols); + intgemm::Int8::PrepareB(B.get(), B_prepared.get(), quant_mult, width, B_cols); AlignedVector C(A_rows * B_cols); // Do the actual multiply. - intgemm::Generic_8bit::Multiply(A_prepared.get(), B_prepared.get(), C.get(), 1.0 / (quant_mult * quant_mult), A_rows, width, B_cols); + intgemm::Int8::Multiply(A_prepared.get(), B_prepared.get(), C.get(), 1.0 / (quant_mult * quant_mult), A_rows, width, B_cols); // Sanity check. C will be row major. assert(fabs(C[0] - top_left_reference) < 0.05); } -- cgit v1.2.3