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

Fixed tsan error. 

PiperOrigin-RevId: 879069355
This commit is contained in:
Ray Smith 2026-03-05 07:59:16 -08:00 committed by Copybara-Service
parent 6721dddf38
commit 79e640a956
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
gemma/flash_attention.cc
View File

@ -1739,7 +1739,8 @@ void ComputeFlashParams(size_t num_tokens, const size_t target_parallelism,
size_t tasks_required = params.back().v_tile_size < k4xNFVTileSize
? k4xNFVTileSize
: kVTileSize;
if (params.back().v_tile_size + tasks_remaining < tasks_required ||
if ((params.back().v_tile_size + tasks_remaining < tasks_required &&
params.back().v_tile_size > 0) ||
params.back().v_tile_size == kVTileSize) {
// We don't have enough tasks remaining to fill a tile, or the
// current tile is full so start new tile.