Home
last modified time | relevance | path

Searched refs:barrier_and_acquire_previous_kernel_writes (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DCUDASymmetricMemoryOps.cu87 barrier_and_acquire_previous_kernel_writes(signal_pads, rank, world_size); in multimem_all_reduce_kernel()
180 barrier_and_acquire_previous_kernel_writes(signal_pads, rank, world_size); in multimem_one_shot_all_reduce_kernel()
H A DCUDASymmetricMemory-inl.h113 __device__ __forceinline__ void barrier_and_acquire_previous_kernel_writes( in barrier_and_acquire_previous_kernel_writes() function