syevd Class — pytorch Architecture
Architecture documentation for the syevd class in CUDASolver.cpp from the pytorch codebase.
Entity Profile
Source Code
aten/src/ATen/native/cuda/linalg/CUDASolver.cpp lines 1129–1143
template <>
void syevd<float>(
cusolverDnHandle_t handle,
cusolverEigMode_t jobz,
cublasFillMode_t uplo,
int n,
float* A,
int lda,
float* W,
float* work,
int lwork,
int* info) {
TORCH_CUSOLVER_CHECK(
cusolverDnSsyevd(handle, jobz, uplo, n, A, lda, W, work, lwork, info));
}
Source
Analyze Your Own Codebase
Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.
Try Supermodel Free