Searched defs:srcRank (Results 1 – 13 of 13) sorted by relevance
171 int srcRank = (rank + ii) % kWorldSize; in oneShotAllReduceKernel() local181 int srcRank = (ii + kWorldSize - rank) % kWorldSize; in oneShotAllReduceKernel() local231 int srcRank = (rank + ii) % kWorldSize; in twoShotAllReduceKernel() local242 int srcRank = (ii + kWorldSize - rank) % kWorldSize; in twoShotAllReduceKernel() local
54 int* srcRank = nullptr; member
196 int srcRank, in recv()
545 int srcRank, in recv()
540 int srcRank, in recv()
872 int srcRank, in recv()
1566 int srcRank, in recv()
4364 int srcRank, in checkForNCCLErrorsInternal()
2827 int srcRank, in recv()
99 int srcRank, in recv()
631 auto srcRank = 0; in testSend() local654 auto srcRank = 1; in testRecv() local
145 worker_id_t srcRank; member
1615 auto srcRank = adaptor.operands()[0].getType().cast<ShapedType>().getRank(); in matchAndRewrite() local