Fix: barrier_wait → mbarrier_wait, barrier_arrive → mbarrier_arrive

2026-05-22 19:48:31 +00:00
parent afe5d1ae21
commit 5f98eaec3a
1 changed files with 2 additions and 2 deletions
--- a/tests/fmha_v3_stage_c_example7.py
+++ b/tests/fmha_v3_stage_c_example7.py
@@ -435,11 +435,11 @@ class FmhaV3StageCMulti:
            cute.arch.fence_view_async_tmem_load()
            # Async-proxy fence so the TMA store sees the SMEM writes.
            cute.arch.fence_proxy("async.shared", space="cta")
-            cute.arch.barrier_arrive(
+            cute.arch.mbarrier_arrive(
                barrier_id=self.epilog_sync_bar_id,
                number_of_threads=32 * len(self.epilogue_warp_id),
            )
-            cute.arch.barrier_wait(
+            cute.arch.mbarrier_wait(
                barrier_id=self.epilog_sync_bar_id,
                number_of_threads=32 * len(self.epilogue_warp_id),
            )