cory-master/BabelStream
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

THRUST Instructions for Spack

Browse Source
This commit is contained in:
Kaan Olgu 2023-06-17 12:05:16 +00:00
parent 0040130b67
commit 217249ff38
1 changed files with 19 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
docs/spack_instructions.md
View File

@ -219,3 +219,21 @@
# Example: # Example:
$ spack install babelstream +tbb $ spack install babelstream +tbb
``` ```
## THRUST
| Flag | Definition |
|-----------| ----------------------------------|
|implementation| Choose one of the implementation for Thrust. Options are `cuda` and `rocm` | `implementation = [cuda]/[rocm]` |
|backend| CUDA's Thrust implementation supports the following backends:- CUDA- OMP - TBB |
| cuda_arch |- List of supported compute capabilities are provided [here](https://github.com/spack/spack/blob/0f271883831bec6da3fc64c92eb1805c39a9f09a/lib/spack/spack/build_systems/cuda.py#LL19C1-L47C6) <br />- Useful [link](https://arnon.dk/matching-sm-architectures-arch-and-gencode-for-various-nvidia-cards/) for matching CUDA gencodes with NVIDIA architectures|
|flags | Additional CUDA flags passed to nvcc, this is appended after `CUDA_ARCH` |
```shell
# Example1: CUDA implementation
$ spack install babelstream +thrust implementation=cuda backend=cuda cuda_arch=<70> flags=<option>
$ spack install babelstream +thrust implementation=cuda backend=omp cuda_arch=<70> flags=<option>
$ spack install babelstream +thrust implementation=cuda backend=tbb cuda_arch=<70> flags=<option>
# Example1: ROCM implementation
* spack install babelstream +thrust implementation=rocm backend=<option>
```