Update nvvm API call from nvvm enum to str (#2985)

2026-01-27 17:28:29 +08:00
parent 7a14467776
commit acb45938e9
29 changed files with 132 additions and 327 deletions
@@ -1224,10 +1224,7 @@ class PersistentDenseGemmKernel:
                            tRS_sC[(None, None, None, c_buffer)],
                        )
                        # Fence and barrier to make sure shared memory store is visible to TMA store
-                        cute.arch.fence_proxy(
-                            cute.arch.ProxyKind.async_shared,
-                            space=cute.arch.SharedSpace.shared_cta,
-                        )
+                        cute.arch.fence_proxy("async.shared", space="cta")
                        epilog_threads = 32 * len(self.epilog_warp_id)
                        cute.arch.barrier(
                            barrier_id=self.epilog_sync_bar_id,