diff options
| -rw-r--r-- | var/spack/repos/builtin/packages/caliper/package.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/caliper/package.py b/var/spack/repos/builtin/packages/caliper/package.py index abaff96c8d..74d8888031 100644 --- a/var/spack/repos/builtin/packages/caliper/package.py +++ b/var/spack/repos/builtin/packages/caliper/package.py @@ -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 @@ class Caliper(CMakePackage, CudaPackage, ROCmPackage): 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 @@ class Caliper(CMakePackage, CudaPackage, ROCmPackage): # ${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) |