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
@@ -1261,10 +1261,7 @@ class GroupedGemmKernel:
                        tRS_sC[(None, None, None, epi_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")
                    self.epilog_sync_barrier.arrive_and_wait()
                    #
                    # store C to global memory with TMA