xgeev Class — pytorch Architecture

Source Code

aten/src/ATen/native/cuda/linalg/CUDASolver.cpp lines 2097–2140

template <>
void xgeev<float>(
    cusolverDnHandle_t handle,
    cusolverDnParams_t params,
    cusolverEigMode_t jobvl,
    cusolverEigMode_t jobvr,
    int64_t n,
    float* A,
    int64_t lda,
    float* W,
    float* VL,
    int64_t ldvl,
    float* VR,
    int64_t ldvr,
    float* bufferOnDevice,
    size_t workspaceInBytesOnDevice,
    float* bufferOnHost,
    size_t workspaceInBytesOnHost,
    int* info) {

  TORCH_CUSOLVER_CHECK(cusolverDnXgeev(
      handle,
      params,
      jobvl,
      jobvr,
      n,
      CUDA_R_32F,
      reinterpret_cast<void*>(A),
      lda,
      CUDA_R_32F,
      reinterpret_cast<void*>(W),
      CUDA_R_32F,
      reinterpret_cast<void*>(VL),
      ldvl,
      CUDA_R_32F,
      reinterpret_cast<void*>(VR),
      ldvr,
      CUDA_R_32F,
      reinterpret_cast<void*>(bufferOnDevice),
      workspaceInBytesOnDevice,
      reinterpret_cast<void*>(bufferOnHost),
      workspaceInBytesOnHost,
      info));
}