Update nvvm API call from nvvm enum to str (#2985)
This commit is contained in:
@@ -1176,10 +1176,7 @@ class SM100PersistentDenseGemmAlphaBetaKernel:
|
||||
tSR_sC[(None, None, None, c_pipeline_consumer_state.index)],
|
||||
tSR_rC,
|
||||
)
|
||||
cute.arch.fence_proxy(
|
||||
cute.arch.ProxyKind.async_shared,
|
||||
space=cute.arch.SharedSpace.shared_cta,
|
||||
)
|
||||
cute.arch.fence_proxy("async.shared", space="cta")
|
||||
c_pipeline.consumer_release(c_pipeline_consumer_state)
|
||||
|
||||
# Advance pipeline states
|
||||
@@ -1206,10 +1203,7 @@ class SM100PersistentDenseGemmAlphaBetaKernel:
|
||||
tiled_copy_r2s, tRS_rD, tRS_sD[(None, None, None, d_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()
|
||||
|
||||
#
|
||||
|
||||
Reference in New Issue
Block a user