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
@@ -1208,10 +1208,7 @@ class PersistentDenseGemmKernel:
                c_buffer = (num_prev_subtiles + subtile_idx) % self.num_c_stage
                cute.copy(tiled_copy_r2s, tRS_rC, 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_sync_barrier.arrive_and_wait()

                #