Files
ROCm/docs/reference/gpu_libraries/linear_algebra.md
2023-03-17 20:47:14 -06:00

2.9 KiB

Linear Algebra Libraries

ROCm libraries for linear algebra are as follows:

:::::{grid} 1 1 2 2 :gutter: 1

:::{grid-item-card} hipBLAS hipBLAS is a compatiblity layer for GPU accelerated BLAS optimized for AMD GPUs via rocBLAS and rocSOLVER. hipBLAS allows for a common interface for other GPU BLAS libraries.

:::

:::{grid-item-card} rocBLAS rocBLAS is an AMD GPU optimized library for BLAS.

:::

:::{grid-item-card} rocSPARSE rocSPARSE is a sparse matrix solver for AMD GPU backends.

:::

:::{grid-item-card} hipSPARSE hipSPARSE is sparse matrix solver library that support AMD and NVIDIA GPU backends.

:::

:::{grid-item-card} rocALUTION rocBLAS is an AMD GPU optimized library for BLAS.

:::

:::{grid-item-card} rocWMMA AMD's C++ library for accelerating mixed-precision matrix multiply-accumulate (MMA) operations leveraging AMD GPU hardware.

:::

:::{grid-item-card} rocSOLVER rocSOLVER is a work-in-progress implementation of a subset of LAPACK functionality on the ROCm platform.

:::

:::{grid-item-card} hipSOLVER hipSOLVER is a LAPACK marshalling library, with multiple supported backends. It sits between the application and a 'worker' LAPACK library, marshalling inputs into the backend library and marshalling results back to the application.

:::

:::::