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

Handle non-threading parallel policy. 

PiperOrigin-RevId: 802012517
This commit is contained in:
Marie White 2025-09-02 00:02:18 -07:00 committed by Copybara-Service
parent 0d2e74d74a
commit 27cb8e12d9
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ops/matmul-inl.h
View File

@ -1147,8 +1147,10 @@ struct MMImpl {
MMPerPackage(A.Extents(), args, config, pkg_idx, range_np)( MMPerPackage(A.Extents(), args, config, pkg_idx, range_np)(
MMNestedParallelPolicy(), A, B, C_rows); MMNestedParallelPolicy(), A, B, C_rows);
break; break;
case ParallelismType::kNone:
case ParallelismType::kSequential: case ParallelismType::kSequential:
MMPerPackage(A.Extents(), args, config, pkg_idx, range_np)(
MMSequentialPolicy(), A, B, C_rows);
case ParallelismType::kNone:
case ParallelismType::kCluster: case ParallelismType::kCluster:
HWY_ABORT("Parallelism type not implemented."); HWY_ABORT("Parallelism type not implemented.");
break; break;