timemory: update to use oneapi packages (#49305)

Signed-off-by: Massimiliano Culpo <massimiliano.culpo@gmail.com>

var/spack/repos/builtin/packages/timemory/package.py

@@ -3,6 +3,7 @@
 # SPDX-License-Identifier: (Apache-2.0 OR MIT)
 #
 # ----------------------------------------------------------------------------
+import os
 
 from spack.package import *
 
@@ -122,7 +123,7 @@ class Timemory(CMakePackage, PythonExtension):
     variant(
         "cpu_target",
         default="auto",
-        description=("Build for specific cpu architecture (specify " "cpu-model)"),
+        description="Build for specific cpu architecture (specify " "cpu-model)",
     )
     variant(
         "use_arch",
@@ -143,7 +144,7 @@ class Timemory(CMakePackage, PythonExtension):
     variant(
         "statistics",
         default=True,
-        description=("Build components w/ support for statistics " "(min/max/stddev)"),
+        description="Build components w/ support for statistics " "(min/max/stddev)",
     )
     variant(
         "extra_optimizations",
@@ -215,7 +216,7 @@ class Timemory(CMakePackage, PythonExtension):
     depends_on("caliper", when="+caliper")
     depends_on("dyninst", when="+dyninst")
     depends_on("gperftools", when="+gperftools")
-    depends_on("intel-parallel-studio", when="+vtune")
+    depends_on("intel-oneapi-vtune", when="+vtune")
     depends_on("arm-forge", when="+allinea_map")
 
     conflicts(
@@ -335,4 +336,26 @@ def cmake_args(self):
             args.append(self.define_from_variant(key, "cuda_arch"))
             args.append(self.define_from_variant("CMAKE_CUDA_STANDARD", "cudastd"))
 
+        if self.spec.satisfies("+vtune"):
+            ittnotify_include = os.path.join(
+                self["intel-oneapi-vtune"].component_prefix, "include"
+            )
+            ittnotify_libraries = find_libraries(
+                "libittnotify",
+                root=self["intel-oneapi-vtune"].component_prefix,
+                shared=False,
+                recursive=True,
+            )
+            if len(ittnotify_libraries) != 1:
+                vtune_spec = self.spec["intel-oneapi-vtune"]
+                raise InstallError(f"{self.spec} cannot find libittnotify from {vtune_spec}")
+
+            ittnotify_library = ittnotify_libraries.libraries[0]
+            args.extend(
+                [
+                    self.define("ITTNOTIFY_INCLUDE_DIR", ittnotify_include),
+                    self.define("ITTNOTIFY_LIBRARY", ittnotify_library),
+                ]
+            )
+
         return args