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
@@ -983,10 +983,7 @@ class DenseGemmKernel:
            c_buffer = 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")
            pipeline.sync(barrier_id=1)

            # TMA store C to global memory