From 5f98eaec3a70b398455f90fdeb328b244f8ffc19 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Fri, 22 May 2026 19:48:31 +0000 Subject: [PATCH] =?UTF-8?q?Fix:=20barrier=5Fwait=20=E2=86=92=20mbarrier=5F?= =?UTF-8?q?wait,=20barrier=5Farrive=20=E2=86=92=20mbarrier=5Farrive?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/fmha_v3_stage_c_example7.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/fmha_v3_stage_c_example7.py b/tests/fmha_v3_stage_c_example7.py index 374e4124..d8b98568 100644 --- 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), )