xsyevBatched_bufferSize Class — pytorch Architecture

Source Code

aten/src/ATen/native/cuda/linalg/CUDASolver.cpp lines 2289–2317

template <>
void xsyevBatched_bufferSize<float>(
    cusolverDnHandle_t handle,
    cusolverDnParams_t params,
    cusolverEigMode_t  jobz,
    cublasFillMode_t uplo,
    int64_t n,
    const float *A,
    int64_t lda,
    const float *W,
    size_t *workspaceInBytesOnDevice,
    size_t *workspaceInBytesOnHost,
    int64_t batchSize) {
  TORCH_CUSOLVER_CHECK(cusolverDnXsyevBatched_bufferSize(
       handle,
       params,
       jobz,
       uplo,
       n,
       CUDA_R_32F,
       reinterpret_cast<const void*>(A),
       lda,
       CUDA_R_32F,
       reinterpret_cast<const void*>(W),
       CUDA_R_32F,
       workspaceInBytesOnDevice,
       workspaceInBytesOnHost,
       batchSize));
}