Searched defs:rank_to_GPU (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/algorithms/quantization/ |
| H A D | test_quantization.py | 214 rank_to_GPU=None, argument
245 rank_to_GPU=None, argument
281 rank_to_GPU=None, argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
| H A D | distributed_test.py | 2009 rank_to_GPU=None, argument
2143 rank_to_GPU=None, argument
2398 rank_to_GPU=None, argument
2528 self, tensor_out, tensor_in, group_id, rank, cuda=True, rank_to_GPU=None argument
2643 rank_to_GPU=None, argument
3038 rank_to_GPU=None, argument
3202 self, group, group_id, rank, cuda=False, rank_to_GPU=None, dtype=torch.float argument
3325 self, group, group_id, rank, cuda=False, rank_to_GPU=None argument
3423 self, group, group_id, rank, cuda=False, rank_to_GPU=None, dtype=torch.float argument
3549 self, tensor_out, tensor_in, group_id, rank, cuda=True, rank_to_GPU=None argument
[all …]
|
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_c10d_nccl.py | 260 def rank_to_GPU(self): member in ProcessGroupNCCLGroupTest
3593 def rank_to_GPU(self): member in NCCLTraceTestBase
4401 def rank_to_GPU(self): member in ProcessGroupNCCLLargerScaleTest
|
| H A D | test_c10d_ops_nccl.py | 58 def rank_to_GPU(self): member in ProcessGroupNCCLOpTest
|