Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DProcessGroupNCCL.cpp2403 int64_t check_gpu_tensors_same_device(const std::vector<at::Tensor>& tensors) { in checkForNCCLErrorsInternal() function
3487 auto total_numel = check_gpu_tensors_same_device(tensors); in checkForNCCLErrorsInternal()