This folder demonstrates cuBLASLMp library API usage.
- Linux
- x86_64
- arm64-sbsa
cuBLASMp is distributed through NVIDIA Developer Zone and also as a part of HPC SDK. cuBLASMp requires CUDA Toolkit, HPC-X, NCCL and GDRCOPY to be installed on the system. The samples require C++11 compatible compiler.
git clone https://github.com/NVIDIA/CUDALibrarySamples.git
cd CUDALibrarySamples/cuBLASMp
mkdir build
cd build
cmake .. -DCMAKE_BUILD_TYPE=Release
make -j
Run examples with mpi command and number of processes according to process grid values, i.e.
mpirun -n 2 ./pgemm
mpirun -n 2 ./ptrsm
mpirun -n 2 ./psyrk
mpirun -n 2 ./pgeadd
mpirun -n 2 ./ptradd