Skip to content

Commit ba2b4c1

Browse files
committed
newer archspec is used so need to set kokkos_arch for nvidia grace
Signed-off-by: laraPPr <[email protected]>
1 parent a0dfa48 commit ba2b4c1

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

eb_hooks.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1025,6 +1025,9 @@ def pre_configure_hook_LAMMPS_zen4_and_Aarch64_cuda(self, *args, **kwargs):
10251025
if cpu_target == CPU_TARGET_ZEN4:
10261026
# There is no support for ZEN4 in LAMMPS yet so falling back to ZEN3
10271027
self.cfg['kokkos_arch'] = 'ZEN3'
1028+
elif cpu_target == CPU_TARGET_NVIDIA_GRACE:
1029+
# There is no support for NVIDA grace in LAMMPS yet so falling back to ARMV81
1030+
self.cfg['kokkos_arch'] 'ARMV81'
10281031

10291032
# Disable SIMD for specific CUDA versions
10301033
if self.version == '2Aug2023_update2':

0 commit comments

Comments
 (0)