File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -146,13 +146,13 @@ We recommend the following modules and environement variables on Jean Zay:
146146
147147.. code-block :: bash
148148
149- -DKokkos_ENABLE_CUDA=ON -DKokkos_ENABLE_VOLTA70 =ON -DKokkos_ENABLE_IMPL_CUDA_MALLOC_ASYNC=OFF
149+ -DKokkos_ENABLE_CUDA=ON -DKokkos_ARCH_VOLTA70 =ON -DKokkos_ENABLE_IMPL_CUDA_MALLOC_ASYNC=OFF
150150
151151 While Ampere A100 GPUs are enabled with
152152
153153.. code-block :: bash
154154
155- -DKokkos_ENABLE_CUDA=ON -DKokkos_ENABLE_AMPERE80 =ON -DKokkos_ENABLE_IMPL_CUDA_MALLOC_ASYNC=OFF
155+ -DKokkos_ENABLE_CUDA=ON -DKokkos_ARCH_AMPERE80 =ON -DKokkos_ENABLE_IMPL_CUDA_MALLOC_ASYNC=OFF
156156
157157 MPI (multi-GPU) can be enabled by adding ``-DIdefix_MPI=ON `` as usual. The malloc async option is here to prevent a bug when using PSM2 with async
158158Cuda malloc possibly leading to openmpi crash or hangs on Jean Zay.
You can’t perform that action at this time.
0 commit comments