rocSPARSE exposes a common interface that provides Basic Linear Algebra Subroutines (BLAS) for sparse computation. It is implemented on top of AMD ROCm runtime and toolchains. rocSPARSE is created using the HIP programming language and optimized for AMD's latest discrete GPUs.
Note
For portability, ROCm provides the hipSPARSE library. hipSPARSE includes a comprehensive, portable interface that supports multiple backends (including rocSPARSE and cuSPARSE). For documentation and examples, see the hipSPARSE documentation.
Note
The published rocSPARSE documentation is available at rocSPARSE in an organized, easy-to-read format, with search and a table of contents. The documentation source files reside in the rocSPARSE/docs folder of this repository. As with all ROCm projects, the documentation is open source. For more information, see Contribute to ROCm documentation.
To build our documentation locally, run the following code:
cd docs pip3 install -r sphinx/requirements.txt python3 -m sphinx -T -E -b html -d _build/doctrees -D language=en . _build/html
Alternatively, build with CMake:
cmake -DBUILD_DOCS=ON ...
- Git
- CMake (3.19 or later)
- ROCm CMake (0.11.0 or later)
- AMD [ROCm] 3.5 platform or later
Optional:
- GoogleTest (required only for tests)
- Use
GTEST_ROOTto specify a location - Alternatively, set
FETCHCONTENT_SOURCE_DIR_GOOGLETESTto a googletest source tree
- Use
- The SuiteSparse test matrices (required only for tests), stored with DVC:
run
dvc pull -R projects/rocsparse/clients/matricesfrom the root of the rocm-libraries repository. DVC needs S3 support, for examplepip install 'dvc[s3]'
By default, CMake doesn't download any of these. Configure with -DALLOW_FETCH_DEPS=ON to download
ROCm CMake, GoogleTest and the test matrices when they're missing.
-
Checkout the rocSPARSE code using either a sparse checkout or a full clone of the rocm-libraries repository.
To limit your local checkout to only the rocSPARSE project, configure
sparse-checkoutbefore cloning. This uses the Git partial clone feature (--filter=blob:none) to reduce how much data is downloaded. Use the following commands for a sparse checkout:git clone --no-checkout --filter=blob:none https://github.com/ROCm/rocm-libraries.git cd rocm-libraries git sparse-checkout init --cone git sparse-checkout set projects/rocsparse .dvc # add projects/rocprim and projects/rocblas to include dependencies git checkout develop # or use the branch you want to work with
To clone the entire rocm-libraries repository, use the following commands. This process takes more time, but is recommended if you want to work with a large number of libraries.
# Clone rocm-libraries, including rocSPARSE, using Git git clone https://github.com/ROCm/rocm-libraries.git # Go to rocSPARSE directory cd rocm-libraries/projects/rocsparse
-
Build rocSPARSE using the
install.shscript.# Run install.sh script # Command line options: # -h|--help - prints help message # -i|--install - install after build # -d|--dependencies - install distro build dependencies (not GoogleTest) # -c|--clients - build library clients too (combines with -i & -d) # -g|--debug - build with debug flag ./install.sh -dci
[!NOTE] Building the tests (
install.sh -c, or-DBUILD_CLIENTS_TESTS=ON) needs GoogleTest and the SuiteSparse test matrices. Install GoogleTest, for example thelibgtest-devpackage, and rundvc pull -R clients/matricesfrom this directory first (in a sparse checkout, rungit sparse-checkout add .dvcbefore that). When you configure with CMake directly,-DALLOW_FETCH_DEPS=ONdownloads both instead. -
Compile rocSPARSE (all compiler specifications are automatically determined).
# Clone rocm-libraries, including rocSPARSE, using Git git clone https://github.com/ROCm/rocm-libraries.git # Go to rocSPARSE directory cd rocm-libraries/projects/rocsparse # Pull the test matrices dvc pull -R clients/matrices # Create and go to the build directory mkdir -p build/release; cd build/release # Configure rocSPARSE # Build options: # BUILD_CLIENTS_TESTS - build tests (OFF) # BUILD_CLIENTS_BENCHMARKS - build benchmarks (OFF) # BUILD_CLIENTS_SAMPLES - build examples (ON) # BUILD_VERBOSE - verbose output (OFF) # BUILD_SHARED_LIBS - build rocSPARSE as a shared library (ON) CXX=/opt/rocm/bin/amdclang++ cmake -DBUILD_CLIENTS_TESTS=ON ../.. # Build make # Install [sudo] make install
To run unit tests, you must build rocSPARSE with -DBUILD_CLIENTS_TESTS=ON.
# Go to rocSPARSE build directory cd rocm-libraries/projects/rocsparse; cd build/release # Run all tests ./clients/staging/rocsparse-test
To run benchmarks, you must build rocSPARSE with -DBUILD_CLIENTS_BENCHMARKS=ON.
# Go to rocSPARSE build directory cd rocm-libraries/projects/rocsparse/build/release # Run benchmark, e.g. ./clients/staging/rocsparse-bench -f hybmv --laplacian-dim 2000 -i 200
To submit an issue, a bug, or a feature request, use the rocm-libraries GitHub issue tracker.
The license file is located in the main repository.