zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

raja: add gpu-profiling variant (#50354)

Browse Source
This commit is contained in:
Victor A. P. Magri 2025-05-08 11:40:56 -07:00 committed by GitHub
parent a82e21e82f
commit f45e312f81
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
var/spack/repos/spack_repo/builtin/packages/raja/package.py
View File

@ -195,6 +195,7 @@ class Raja(CachedCMakePackage, CudaPackage, ROCmPackage):
)
variant("omptarget", default=False, description="Build OpenMP on target device support")
variant("sycl", default=False, description="Build sycl backend")
variant("gpu-profiling", default=False, description="Enable GPU profiling")
variant("plugins", default=False, description="Enable runtime plugins")
variant("examples", default=True, description="Build examples.")
@ -267,6 +268,10 @@ class Raja(CachedCMakePackage, CudaPackage, ROCmPackage):
for sm_ in CudaPackage.cuda_arch_values:
depends_on("camp +cuda cuda_arch={0}".format(sm_), when="cuda_arch={0}".format(sm_))
conflicts("+gpu-profiling", when="~cuda~rocm", msg="GPU profiling requires CUDA or ROCm")
conflicts("+gpu-profiling +cuda", when="@:2022.02.99")
conflicts("+gpu-profiling +rocm", when="@:2022.02.99")
conflicts("+omptarget +rocm")
conflicts("+sycl +omptarget")
conflicts("+sycl +rocm")
@ -323,11 +328,7 @@ def initconfig_hardware_entries(self):
entries.append("#------------------{0}\n".format("-" * 30))
entries.append(cmake_cache_option("ENABLE_OPENMP", spec.satisfies("+openmp")))
if spec.satisfies("+cuda"):
entries.append(cmake_cache_option("ENABLE_CUDA", True))
else:
entries.append(cmake_cache_option("ENABLE_CUDA", False))
entries.append(cmake_cache_option("ENABLE_CUDA", spec.satisfies("+cuda")))
if spec.satisfies("+rocm"):
entries.append(cmake_cache_option("ENABLE_HIP", True))
@ -376,6 +377,12 @@ def initconfig_package_entries(self):
)
entries.append(cmake_cache_option("RAJA_ENABLE_SYCL", spec.satisfies("+sycl")))
entries.append(
cmake_cache_option("RAJA_ENABLE_NV_TOOLS_EXT", spec.satisfies("+gpu-profiling +cuda"))
)
entries.append(
cmake_cache_option("RAJA_ENABLE_ROCTX", spec.satisfies("+gpu-profiling +rocm"))
)
if spec.satisfies("+lowopttest"):
entries.append(cmake_cache_string("CMAKE_CXX_FLAGS_RELEASE", "-O1"))