Skip to content

Commit

Permalink
fix lambda extension for nvcc
Browse files Browse the repository at this point in the history
  • Loading branch information
SimeonEhrig committed Jan 8, 2025
1 parent 1aedc1a commit 5a6f06b
Showing 1 changed file with 8 additions and 4 deletions.
12 changes: 8 additions & 4 deletions script/job_generator/generate_job_yaml.py
Original file line number Diff line number Diff line change
Expand Up @@ -345,15 +345,19 @@ def job_variables(job: Dict[str, Tuple[str, str]]) -> Dict[str, str]:
variables["ALPAKA_CI_STDLIB"] = "libstdc++"
variables["CMAKE_CUDA_ARCHITECTURES"] = job[SM_LEVEL][VERSION]
variables["ALPAKA_CI_CUDA_VERSION"] = job[ALPAKA_ACC_GPU_CUDA_ENABLE][VERSION]
variables["alpaka_RELOCATABLE_DEVICE_CODE"] = OFF
variables["alpaka_CUDA_SHOW_REGISTER"] = OFF
variables["alpaka_CUDA_KEEP_FILES"] = OFF
variables["alpaka_CUDA_EXPT_EXTENDED_LAMBDA"] = OFF
variables["alpaka_RELOCATABLE_DEVICE_CODE"] = "OFF"
variables["alpaka_CUDA_SHOW_REGISTER"] = "OFF"
variables["alpaka_CUDA_KEEP_FILES"] = "OFF"
variables["alpaka_CUDA_EXPT_EXTENDED_LAMBDA"] = "OFF"

if job[DEVICE_COMPILER][NAME] == NVCC:
# general configuration, if nvcc is the CUDA compiler
variables["ALPAKA_CI_CUDA_COMPILER"] = "nvcc"

# MdSpan requires alpaka_CUDA_EXPT_EXTENDED_LAMBDA
if job[MDSPAN][VERSION] == ON_VER:
variables["alpaka_CUDA_EXPT_EXTENDED_LAMBDA"] = "ON"

# configuration, if GCC is the CUDA host compiler
if job[HOST_COMPILER][NAME] == GCC:
variables["ALPAKA_CI_CXX"] = "g++"
Expand Down

0 comments on commit 5a6f06b

Please sign in to comment.