From a4484c3afedd8d03f8b7d688e6db20d79902ee1c Mon Sep 17 00:00:00 2001
From: Matthias Cremon <matthiascremon@meta.com>
Date: Wed, 9 Oct 2024 09:16:31 -0700
Subject: [PATCH] Allow int8 type in quantized_matmul (#5898)

Summary:

As titled.

Reviewed By: zonglinpeng

Differential Revision: D63659947
---
 backends/cadence/reference/operators/quantized_matmul_out.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/backends/cadence/reference/operators/quantized_matmul_out.cpp b/backends/cadence/reference/operators/quantized_matmul_out.cpp
index b0a9393cd01..5e1a49c0b5a 100644
--- a/backends/cadence/reference/operators/quantized_matmul_out.cpp
+++ b/backends/cadence/reference/operators/quantized_matmul_out.cpp
@@ -49,7 +49,7 @@ __attribute__((noinline)) void qmatmul(
           sum += (X[i * n + k] - X_zero_point) * (y[k * p + j] - Y_zero_point);
         }
       }
-      Z[i * p + j] = kernels::quantize<uint8_t>(sum, Z_scale, Z_zero_point);
+      Z[i * p + j] = kernels::quantize<TZ>(sum, Z_scale, Z_zero_point);
     }
   }
 }