diff options
author | Benoit Jacob <benoitjacob@google.com> | 2020-12-22 00:53:34 +0300 |
---|---|---|
committer | Copybara-Service <copybara-worker@google.com> | 2020-12-22 00:53:53 +0300 |
commit | 878283640de7946a43053e8ebf4f15114fbc9156 (patch) | |
tree | 0215a3ef8a24dc3c9dd6370a91c60b8081f3385d | |
parent | c162e5d936989ab8d339857698c07d88d164c1e8 (diff) |
Fix the new raw accumulators example - being raw accumulators, it's not 'per channel', as there is no multiplier here.
PiperOrigin-RevId: 348522764
-rw-r--r-- | example/example.cc | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/example/example.cc b/example/example.cc index 4e81304..3bb95f4 100644 --- a/example/example.cc +++ b/example/example.cc @@ -126,7 +126,7 @@ void ExampleMulInt8PerChannelQuantized(ruy::Context *context) { std::cout << "RHS:\n" << rhs; std::cout << "Result:\n" << dst << "\n"; } -void ExampleMulInt8PerChannelGetRawAccumulators(ruy::Context *context) { +void ExampleMulInt8GetRawAccumulators(ruy::Context *context) { const std::int8_t lhs_data[] = {1, 2, 3, 4}; const std::int8_t rhs_data[] = {1, 2, 3, 4}; std::int32_t dst_data[4]; @@ -145,8 +145,7 @@ void ExampleMulInt8PerChannelGetRawAccumulators(ruy::Context *context) { ruy::MulParams<std::int32_t, std::int32_t> mul_params; ruy::Mul(lhs, rhs, mul_params, context, &dst); - std::cout << "Example Mul, int8 quantized with per-channel multipliers, " - "returning raw int32 accumulators:\n"; + std::cout << "Example Mul, returning raw int32 accumulators:\n"; std::cout << "LHS:\n" << lhs; std::cout << "RHS:\n" << rhs; std::cout << "Result:\n" << dst << "\n"; @@ -158,5 +157,5 @@ int main() { ExampleMulFloatWithBiasAddAndClamp(&context); ExampleMulUint8AsymmetricQuantized(&context); ExampleMulInt8PerChannelQuantized(&context); - ExampleMulInt8PerChannelGetRawAccumulators(&context); + ExampleMulInt8GetRawAccumulators(&context); } |