_rocsolver_syevd_strided_batched Class — pytorch Architecture

Source Code

aten/src/ATen/native/cuda/linalg/BatchLinearAlgebraLib.cpp lines 1210–1225

template <typename scalar_t>
rocblas_status _rocsolver_syevd_strided_batched(
    rocblas_handle handle,
    const rocblas_evect evect,
    const rocblas_fill uplo,
    const rocblas_int n,
    scalar_t* A,
    const rocblas_int lda,
    const rocblas_stride strideA,
    scalar_t* D,
    const rocblas_stride strideD,
    scalar_t* E,
    const rocblas_stride strideE,
    rocblas_int* info,
    const rocblas_int batch_count
);