Skip to content

Commit

Permalink
Merge branch 'ykim/machinefiles/fix-nvidiagpu-pm-gpu' into next (PR #…
Browse files Browse the repository at this point in the history 
…6460)

For pm-gpu: Update CMAKE_EXE_LINKER_FLAGS Cmake variable to use Nvidia cc80 compute capability
to match with Nvidia A100 GPUs
Should only affect ACC
  • Loading branch information
@ndkeen
ndkeen committed Jun 13, 2024
2 parents 4122700 + 578f1bd commit e7b4333
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 4 deletions.
5 changes: 3 additions & 2 deletions cime_config/machines/cmake_macros/nvidiagpu_muller-gpu.cmake
View file
Original file line number Diff line number Diff line change
Expand Up @@ -6,8 +6,9 @@ if (COMP_NAME STREQUAL gptl)
endif()
string(APPEND CPPDEFS " -DTHRUST_IGNORE_CUB_VERSION_CHECK")
string(APPEND CMAKE_CUDA_FLAGS " -ccbin CC -O2 -arch sm_80 --use_fast_math")
string(APPEND CMAKE_EXE_LINKER_FLAGS " -acc -gpu=cc70,cc60 -Minfo=accel")
set(CMAKE_CUDA_ARCHITECTURES "80")
string(APPEND CMAKE_Fortran_FLAGS " -acc -gpu=cc80 -Minfo=accel")
string(APPEND CMAKE_EXE_LINKER_FLAGS " -acc -gpu=cc80 -Minfo=accel")
set(SCC "cc")
set(SCXX "CC")
set(SFC "ftn")
string(APPEND CMAKE_Fortran_FLAGS " -acc -gpu=cc70,cc60 -Minfo=accel")
5 changes: 3 additions & 2 deletions cime_config/machines/cmake_macros/nvidiagpu_pm-gpu.cmake
View file
Original file line number Diff line number Diff line change
Expand Up @@ -6,8 +6,9 @@ if (COMP_NAME STREQUAL gptl)
endif()
string(APPEND CPPDEFS " -DTHRUST_IGNORE_CUB_VERSION_CHECK")
string(APPEND CMAKE_CUDA_FLAGS " -ccbin CC -O2 -arch sm_80 --use_fast_math")
string(APPEND CMAKE_EXE_LINKER_FLAGS " -acc -gpu=cc70,cc60 -Minfo=accel")
set(CMAKE_CUDA_ARCHITECTURES "80")
string(APPEND CMAKE_Fortran_FLAGS " -acc -gpu=cc80 -Minfo=accel")
string(APPEND CMAKE_EXE_LINKER_FLAGS " -acc -gpu=cc80 -Minfo=accel")
set(SCC "cc")
set(SCXX "CC")
set(SFC "ftn")
string(APPEND CMAKE_Fortran_FLAGS " -acc -gpu=cc70,cc60 -Minfo=accel")

0 comments on commit e7b4333

Please sign in to comment.