Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DProcessGroupNCCL.cpp2376 void check_gpu_single_tensor( in checkForNCCLErrorsInternal() function
3449 check_gpu_single_tensor(tensor); in checkForNCCLErrorsInternal()
3543 check_gpu_single_tensor(tensor); in checkForNCCLErrorsInternal()
3644 check_gpu_single_tensor(tensor); in checkForNCCLErrorsInternal()
3736 check_gpu_single_tensor(inputTensor); in checkForNCCLErrorsInternal()
3864 check_gpu_single_tensor(outputTensor); in checkForNCCLErrorsInternal()
4154 check_gpu_single_tensor(outputTensor, true); in checkForNCCLErrorsInternal()
4155 check_gpu_single_tensor(inputTensor, true); in checkForNCCLErrorsInternal()
4267 check_gpu_single_tensor(outputTensors[r], true); in checkForNCCLErrorsInternal()
4268 check_gpu_single_tensor(inputTensors[r], true); in checkForNCCLErrorsInternal()
[all …]