Searched defs:num_ranks (Results 1 – 3 of 3) sorted by relevance
32 def checkAllgather(self, num_ranks, all_gathered, local_gathered): argument
92 TestCase* MakeTestCase(int num_ranks, ncclRedOp_t reduction_op, in MakeTestCase()188 const int num_ranks = 3; in TEST_F() local217 const int num_ranks = 1; // 2; in TEST_F() local
258 __u32 num_ranks; /* from KFD */ member