fix compile error

This commit is contained in:
Feng Shijie
2025-07-30 04:52:08 +00:00
parent c117a1986a
commit 2e5d4c74cd

View File

@@ -291,7 +291,7 @@ struct FlatmmKernel
<< ", persistent_block_size: " << persistent_block_size
<< ", total_work_tile_cnt: " << total_work_tile_cnt << std::endl;
assert(KBatch == 1);
assert(kargs.k_batch == 1);
return dim3(min(persistent_block_size, total_work_tile_cnt), 1, kargs.k_batch);
}
else