Searched refs:check_gpu_single_tensor (Results 1 – 1 of 1) sorted by relevance
2376 void check_gpu_single_tensor( in checkForNCCLErrorsInternal() function3449 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 …]