happyz
/
gemma.cpp
mirror of https://github.com/google/gemma.cpp.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix for GenerateZeroMat call in TestTiledMatMul 

PiperOrigin-RevId: 640180868
This commit is contained in:
Phil Culliton 2024-06-04 09:31:42 -07:00 committed by Copybara-Service
parent 9e213b3d96
commit e71d82ead9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
gemma/ops_test.cc
View File

@ -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);