getrfBatched Class — pytorch Architecture
Architecture documentation for the getrfBatched class in CUDABlas.cpp from the pytorch codebase.
Entity Profile
Source Code
aten/src/ATen/cuda/CUDABlas.cpp lines 2767–2773
template <>
void getrfBatched<double>(
int n, double** dA_array, int ldda, int* ipiv_array, int* info_array, int batchsize) {
auto handle = at::cuda::getCurrentCUDABlasHandle();
TORCH_CUDABLAS_CHECK(cublasDgetrfBatched(
handle, n, dA_array, ldda, ipiv_array, info_array, batchsize));
}
Source
Analyze Your Own Codebase
Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.
Try Supermodel Free