From f9a5f76b7a4f4298d4ec4039c9ae2a673c9fab93 Mon Sep 17 00:00:00 2001 From: aragorn-guan Date: Sat, 14 Feb 2026 17:51:20 +0800 Subject: [PATCH] Replace fence proxy to the latest routine code in examples/distributed/all_reduce_tma.py (#3027) --- examples/python/CuTeDSL/distributed/all_reduce_tma.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/examples/python/CuTeDSL/distributed/all_reduce_tma.py b/examples/python/CuTeDSL/distributed/all_reduce_tma.py index b3496c2b..35a01212 100644 --- a/examples/python/CuTeDSL/distributed/all_reduce_tma.py +++ b/examples/python/CuTeDSL/distributed/all_reduce_tma.py @@ -415,10 +415,7 @@ class AllReduceTmaKernel: # ====================================================================== if warp_idx == 0: # Fence to ensure SMEM writes are visible - cute.arch.fence_proxy( - cute.arch.ProxyKind.async_shared, - space=cute.arch.SharedSpace.shared_cta, - ) + cute.arch.fence_proxy("async.shared", space="cta") smem_tile_out = cute.slice_(staged_smem_tensor, (None, 0))