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

caliper: add new variant to support Intel Vtune (#44147)

Browse Source
This commit is contained in:
Pramod Kumbhar 2024-05-20 18:27:42 +02:00 committed by GitHub
parent a28405700e
commit eb70c9f5b9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
var/spack/repos/builtin/packages/caliper/package.py
View File

@ -94,6 +94,7 @@ class Caliper(CMakePackage, CudaPackage, ROCmPackage):
variant("sosflow", default=False, description="Enable SOSflow support")
variant("fortran", default=False, description="Enable Fortran support")
variant("variorum", default=False, description="Enable Variorum support")
variant("vtune", default=False, description="Enable Intel Vtune support")
variant("kokkos", default=True, when="@2.3.0:", description="Enable Kokkos profiling support")
variant("tests", default=False, description="Enable tests")
@ -109,6 +110,7 @@ class Caliper(CMakePackage, CudaPackage, ROCmPackage):
depends_on("unwind@1.2:1", when="+libunwind")
depends_on("elfutils", when="+libdw")
depends_on("variorum", when="+variorum")
depends_on("intel-oneapi-vtune", when="+vtune")
depends_on("sosflow@spack", when="@1.0:1+sosflow")
@ -149,6 +151,7 @@ def cmake_args(self):
self.define_from_variant("WITH_ROCTRACER", "rocm"),
self.define_from_variant("WITH_ROCTX", "rocm"),
self.define_from_variant("WITH_VARIORUM", "variorum"),
self.define_from_variant("WITH_VTUNE", "vtune"),
self.define_from_variant("WITH_KOKKOS", "kokkos"),
]
@ -181,6 +184,10 @@ def cmake_args(self):
# ${CUDA_TOOLKIT_ROOT_DIR}/extras/CUPTI
args.append("-DCUPTI_PREFIX=%s" % spec["cuda"].prefix)
if "+vtune" in spec:
itt_dir = join_path(spec["intel-oneapi-vtune"].prefix, "vtune", "latest")
args.append("-DITT_PREFIX=%s" % itt_dir)
if "+rocm" in spec:
args.append("-DCMAKE_CXX_COMPILER={0}".format(spec["hip"].hipcc))
args.append("-DROCM_PREFIX=%s" % spec["hsa-rocr-dev"].prefix)