3.8 KiB
Linear Algebra Libraries
ROCm libraries for linear algebra are as follows:
:::::{grid} 1 1 2 2 :gutter: 1
:::{grid-item-card} rocBLAS
rocBLAS is an AMD GPU optimized library for BLAS (Basic Linear Algebra Subprograms).
:::
:::{grid-item-card} hipBLAS
hipBLAS is a compatibility 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} hipBLASLt
hipBLASLt is a library that provides general matrix-matrix operations with a
flexible API and extends functionalities beyond traditional BLAS library.
hipBLASLt is exposed APIs in HIP programming language with an underlying
optimized generator as a back-end kernel provider.
:::
:::{grid-item-card} rocALUTION
rocALUTION is a sparse linear algebra library with focus on exploring
fine-grained parallelism on top of AMD's ROCm runtime and toolchains, targeting
modern CPU and GPU platforms.
:::
:::{grid-item-card} rocWMMA
rocWMMA provides an API to break down mixed precision matrix multiply-accumulate
(MMA) problems into fragments and distributes these over GPU wavefronts.
:::
:::{grid-item-card} rocSOLVER
rocSOLVER provides a subset of LAPACK (Linear Algebra Package) functionality on the ROCm platform.
:::
:::{grid-item-card} hipSOLVER
hipSOLVER is a LAPACK marshalling library supporting both rocSOLVER and cuSOLVER
as backends whilst exporting a unified interface.
:::
:::{grid-item-card} rocSPARSE
rocSPARSE is a library to provide BLAS for sparse computations.
:::
:::{grid-item-card} hipSPARSE
hipSPARSE is a marshalling library to provide sparse BLAS functionality,
supporting both rocSPARSE and cuSPARSE as backends.
:::
:::::