Searched refs:num_ranks (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager_test.cc | 468 const int num_ranks = this->NumGPUs(); in TYPED_TEST() local
473 this->MakeReductionTestCase(/*num_nodes=*/1, num_ranks, reduction_op, in TYPED_TEST()
475 for (int rank = 0; rank < num_ranks; ++rank) { in TYPED_TEST()
476 auto* device = this->GetDevice(num_ranks, /*node=*/0, rank); in TYPED_TEST()
486 {"allreduce", /*num_local_devices=*/num_ranks, in TYPED_TEST()
487 /*num_global_devices=*/num_ranks, /*communicator_key=*/"", in TYPED_TEST()
503 const int num_ranks = this->NumGPUs(); in TYPED_TEST() local
515 /*num_nodes=*/1, num_ranks, ncclSum, in TYPED_TEST()
517 for (int j = 0; j < num_ranks; ++j) { in TYPED_TEST()
522 for (int rank = 0; rank < num_ranks; ++rank) { in TYPED_TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_test.cc | 110 void Init(const int num_ranks, const int instance_key) { in Init() argument
115 CHECK_LE(num_ranks, gpus_.size()); in Init()
116 for (int rank = 0; rank < num_ranks; ++rank) { in Init()
133 const int group_key = num_ranks; in Init()
136 col_params_->group.group_size = num_ranks; in Init()
142 col_params_->group.num_devices_per_task[task_name] = num_ranks; in Init()
143 for (int rank = 0; rank < num_ranks; ++rank) { in Init()
147 for (int rank = 0; rank < num_ranks; ++rank) { in Init()
158 const int tensor_length, const int num_ranks) = 0;
185 void RunTest(int num_ranks, int input_length, int instance_key) { in RunTest() argument
[all …]
|
| /external/kernel-headers/original/uapi/linux/ |
| D | kfd_ioctl.h | 323 __u32 num_ranks; /* from KFD */ member
|