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

timemory: update to use oneapi packages (#49305)

Browse Source 
Signed-off-by: Massimiliano Culpo <massimiliano.culpo@gmail.com>
This commit is contained in:
Massimiliano Culpo 2025-03-17 08:48:57 +01:00 committed by GitHub
parent 03b864f986
commit 461f1d186b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 26 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
var/spack/repos/builtin/packages/timemory/package.py
View File

@ -3,6 +3,7 @@
# SPDX-License-Identifier: (Apache-2.0 OR MIT) # SPDX-License-Identifier: (Apache-2.0 OR MIT)
# #
# ---------------------------------------------------------------------------- # ----------------------------------------------------------------------------
import os
from spack.package import * from spack.package import *
@ -122,7 +123,7 @@ class Timemory(CMakePackage, PythonExtension):
variant( variant(
"cpu_target", "cpu_target",
default="auto", default="auto",
description=("Build for specific cpu architecture (specify " "cpu-model)"), description="Build for specific cpu architecture (specify " "cpu-model)",
) )
variant( variant(
"use_arch", "use_arch",
@ -143,7 +144,7 @@ class Timemory(CMakePackage, PythonExtension):
variant( variant(
"statistics", "statistics",
default=True, default=True,
description=("Build components w/ support for statistics " "(min/max/stddev)"), description="Build components w/ support for statistics " "(min/max/stddev)",
) )
variant( variant(
"extra_optimizations", "extra_optimizations",
@ -215,7 +216,7 @@ class Timemory(CMakePackage, PythonExtension):
depends_on("caliper", when="+caliper") depends_on("caliper", when="+caliper")
depends_on("dyninst", when="+dyninst") depends_on("dyninst", when="+dyninst")
depends_on("gperftools", when="+gperftools") 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") depends_on("arm-forge", when="+allinea_map")
conflicts( conflicts(
@ -335,4 +336,26 @@ def cmake_args(self):
args.append(self.define_from_variant(key, "cuda_arch")) args.append(self.define_from_variant(key, "cuda_arch"))
args.append(self.define_from_variant("CMAKE_CUDA_STANDARD", "cudastd")) 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 return args