Fix the racing condition of mixed-input gemm when writing the registers (#1931)

* move two warpgroup_wait

* merge main

---------

Co-authored-by: Siyuan Fu <siyuanf@nvidia.com>
This commit is contained in:
Lain
2024-11-08 13:15:54 -05:00
committed by GitHub
co-authored by Siyuan Fu
parent d656afbd2a
commit 8aa95dbb88
2 changed files with 4 additions and 3 deletions
@@ -724,4 +724,4 @@ int main(int argc, char const **args) {
return 0;
}
/////////////////////////////////////////////////////////////////////////////////////////////////
/////////////////////////////////////////////////////////////////////////////////////////////////