Home
last modified time | relevance | path

Searched refs:nccl_params (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_collective_thunk.cc176 LockNcclComm(params.nccl_params, config().replica_groups, in ExecuteOnStream()
198 const NcclExecuteParams& nccl_params) { in GetDeviceString() argument
199 int device_ordinal = nccl_params.stream->parent()->device_ordinal(); in GetDeviceString()
201 nccl_params.GetGlobalDeviceId().ValueOrDie(); in GetDeviceString()
203 nccl_params.device_assn->LogicalIdForDevice(global_device_id) in GetDeviceString()
Dreplica_id_thunk.cc25 params.nccl_params.GetGlobalDeviceId()); in ExecuteOnStream()
28 params.nccl_params.device_assn->LogicalIdForDevice(global_device_id)); in ExecuteOnStream()
Dnccl_collective_permute_thunk.cc123 params.nccl_params.GetGlobalDeviceId()); in RunNcclCollective()
126 params.nccl_params.device_assn->LogicalIdForDevice(global_device_id)); in RunNcclCollective()
131 std::string device_string = GetDeviceString(params.nccl_params); in RunNcclCollective()
Dthunk.h114 NcclExecuteParams nccl_params; member
Dthunk.cc28 nccl_params(run_options, stream) {} in ExecuteParams()