Searched defs:wrapper_pg (Results 1 – 1 of 1) sorted by relevance
72 def _test_collective_hang(self, wrapper_pg, use_cuda=False): argument93 def _test_collectives_op_mismatch(self, wrapper_pg, use_cuda=False): argument148 def _test_collective_shape_mismatch(self, wrapper_pg, use_cuda=False): argument293 def _test_nccl_only_op_mismatch(self, wrapper_pg): argument311 def _test_nccl_only_shape_mismatch(self, wrapper_pg): argument