tilelang.contrib.cutedsl.mbar ============================= .. py:module:: tilelang.contrib.cutedsl.mbar .. autoapi-nested-parse:: Simple wrappers that delegate to cutlass.cute.arch implementations. We use the existing implementations from cutlass rather than reinventing the wheel. Functions --------- .. autoapisummary:: tilelang.contrib.cutedsl.mbar.mbarrier_wait tilelang.contrib.cutedsl.mbar.mbarrier_cp_async_arrive tilelang.contrib.cutedsl.mbar.fence_proxy_async tilelang.contrib.cutedsl.mbar.fence_barrier_init Module Contents --------------- .. py:function:: mbarrier_wait(mbar_ptr, phase, timeout_ns = 10000000, *, loc=None, ip=None) Waits on a mbarrier with a specified phase. .. py:function:: mbarrier_cp_async_arrive(mbar_ptr, *, loc=None, ip=None) .. py:function:: fence_proxy_async() .. py:function:: fence_barrier_init()