From e71d82ead95206595f3137c2c38a6d297f14381d Mon Sep 17 00:00:00 2001
From: Phil Culliton <philculliton@google.com>
Date: Tue, 4 Jun 2024 09:31:42 -0700
Subject: [PATCH] Fix for GenerateZeroMat call in TestTiledMatMul

PiperOrigin-RevId: 640180868
---
 gemma/ops_test.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gemma/ops_test.cc b/gemma/ops_test.cc
index 6425890..dd0b052 100644
--- a/gemma/ops_test.cc
+++ b/gemma/ops_test.cc
@@ -484,7 +484,7 @@ void TestTiledMatMul() {
 
   CompressedArray<MatT, kM * kN> a = GenerateMat<MatT, kM, kN>(0, pool);
   CompressedArray<MatT, kN * kK> b = GenerateMat<MatT, kN, kK>(0, pool);
-  CompressedArray<float, kN * kK> c_slow = GenerateZeroMat<float, kM, kK>(pool);
+  CompressedArray<float, kM * kK> c_slow = GenerateZeroMat<float, kM, kK>(pool);
   MatMulSlow<kM, kN, kK>(a.data(), b.data(), c_slow.data());
 
   hwy::AlignedFreeUniquePtr<float[]> c = hwy::AllocateAligned<float>(kM * kK);