xgeev_bufferSize Class — pytorch Architecture

Source Code

aten/src/ATen/native/cuda/linalg/CUDASolver.cpp lines 1960–1992

template <>
void xgeev_bufferSize<float>(
    cusolverDnHandle_t handle,
    cusolverDnParams_t params,
    cusolverEigMode_t jobvl,
    cusolverEigMode_t jobvr,
    int64_t n,
    const float* A,
    int64_t lda,
    const float* W,
    const float* VL,
    int64_t ldvl,
    const float* VR,
    int64_t ldvr,
    size_t* workspaceInBytesOnDevice,
    size_t* workspaceInBytesOnHost) {
  TORCH_CUSOLVER_CHECK(cusolverDnXgeev_bufferSize(
      handle, params, jobvl, jobvr, n,
      CUDA_R_32F,
      reinterpret_cast<const void*>(A),
      lda,
      CUDA_R_32F,
      reinterpret_cast<const void*>(W),
      CUDA_R_32F,
      reinterpret_cast<const void*>(VL),
      ldvl,
      CUDA_R_32F,
      reinterpret_cast<const void*>(VR),
      ldvr,
      CUDA_R_32F,
      workspaceInBytesOnDevice,
      workspaceInBytesOnHost));
}