Searched refs:root_rank (Results 1 – 1 of 1) sorted by relevance
163 int root_rank = -1; member516 CHECK_EQ(collective->root_rank, -1); in RunCollective()517 collective->root_rank = rank; in RunCollective()522 CHECK_NE(collective->root_rank, -1); in RunCollective()588 collective->root_rank, nccl_comm, *cu_stream); in LoopKernelLaunches()598 collective->root_rank, nccl_comm, *cu_stream); in LoopKernelLaunches()