Home
last modified time | relevance | path

Searched defs:signal_pads (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DCUDASymmetricMemory.cu273 std::vector<void*> signal_pads, in CUDASymmetricMemory()
425 uint32_t** signal_pads, in barrier_kernel()
448 uint32_t** signal_pads, in put_signal_kernel()
471 uint32_t** signal_pads, in wait_signal_kernel()
742 std::vector<void*> signal_pads(world_size, nullptr); in rendezvous() local
H A DCUDASymmetricMemory-inl.h89 uint32_t** signal_pads, in barrier()
114 uint32_t** signal_pads, in barrier_and_acquire_previous_kernel_writes()
H A DCUDASymmetricMemoryOps.cu81 uint32_t** signal_pads, in multimem_all_reduce_kernel()
174 uint32_t** signal_pads, in multimem_one_shot_all_reduce_kernel()