Searched refs:check_gpu_tensors_same_device (Results 1 – 1 of 1) sorted by relevance
2403 int64_t check_gpu_tensors_same_device(const std::vector<at::Tensor>& tensors) { in checkForNCCLErrorsInternal() function3487 auto total_numel = check_gpu_tensors_same_device(tensors); in checkForNCCLErrorsInternal()