From de8221ea58a86ff5feb0446ba7505dc5fe371bb5 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Sat, 23 May 2026 02:45:30 +0000 Subject: [PATCH] fix: correct bSG_gC indexing (6 modes) --- tests/unit/test_fmha_v3_stage_c.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/test_fmha_v3_stage_c.py b/tests/unit/test_fmha_v3_stage_c.py index 6b60e81b..7e7aa8b8 100644 --- a/tests/unit/test_fmha_v3_stage_c.py +++ b/tests/unit/test_fmha_v3_stage_c.py @@ -392,7 +392,7 @@ class FmhaV3StageCMulti: num_threads=32 * len(self.epilogue_warp_id), ) epi_bar.arrive_and_wait() - cute.copy(tma_c, bSG_sC[(None, 0)], bSG_gC[(None, 0, 0)]) + cute.copy(tma_c, bSG_sC[(None, 0)], bSG_gC[(None, 0, 0, 0, 0, 0)]) cute.arch.cp_async_bulk_commit_group() cute.arch.cp_async_bulk_wait_group(0, read=True)