xsyevBatched Class — pytorch Architecture

Source Code

aten/src/ATen/native/cuda/linalg/CUDASolver.cpp lines 2409–2443

template <>
void xsyevBatched<float>(
    cusolverDnHandle_t handle,
    cusolverDnParams_t params,
    cusolverEigMode_t jobz,
    cublasFillMode_t uplo,
    int64_t n,
    float *A,
    int64_t lda,
    float *W,
    void *bufferOnDevice,
    size_t workspaceInBytesOnDevice,
    void *bufferOnHost,
    size_t workspaceInBytesOnHost,
    int *info,
    int64_t batchSize) {
  TORCH_CUSOLVER_CHECK(cusolverDnXsyevBatched(
       handle,
       params,
       jobz,
       uplo,
       n,
       CUDA_R_32F,
       reinterpret_cast<void*>(A),
       lda,
       CUDA_R_32F,
       reinterpret_cast<void*>(W),
       CUDA_R_32F,
       bufferOnDevice,
       workspaceInBytesOnDevice,
       bufferOnHost,
       workspaceInBytesOnHost,
       info,
       batchSize));
}