213 lines
8.7 KiB
Python
213 lines
8.7 KiB
Python
# Copyright 2013-2024 Lawrence Livermore National Security, LLC and other
|
|
# Spack Project Developers. See the top-level COPYRIGHT file for details.
|
|
#
|
|
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
|
|
#
|
|
# ----------------------------------------------------------------------------
|
|
|
|
from spack.package import *
|
|
|
|
|
|
class Omnitrace(CMakePackage):
|
|
"""Application Profiling, Tracing, and Analysis"""
|
|
|
|
homepage = "https://rocm.docs.amd.com/projects/omnitrace/en/latest/index.html"
|
|
git = "https://github.com/ROCm/omnitrace.git"
|
|
url = "https://github.com/ROCm/omnitrace/archive/refs/tags/rocm-6.2.0.tar.gz"
|
|
maintainers("dgaliffiAMD", "afzpatel", "srekolam", "renjithravindrankannath", "jrmadsen")
|
|
|
|
license("MIT")
|
|
|
|
version("amd-mainline", branch="amd-mainline", submodules=True)
|
|
version("amd-staging", branch="amd-staging", submodules=True)
|
|
version(
|
|
"1.12.0", tag="v1.12.0", commit="abff23ac4238da6d7891d9ac9f36a919e30bf759", submodules=True
|
|
)
|
|
version(
|
|
"rocm-6.2.4",
|
|
tag="rocm-6.2.4",
|
|
commit="47597c1be3699c5aaaf6164061ee4189c6b32445",
|
|
submodules=True,
|
|
)
|
|
version(
|
|
"rocm-6.2.1",
|
|
tag="rocm-6.2.1",
|
|
commit="df91a342370401c93b5278bf082e520d6a0e22e9",
|
|
submodules=True,
|
|
)
|
|
version(
|
|
"1.11.4", tag="v1.11.4", commit="6b0627f5b7d4b05c3b7b1da581e474d48ebe36cf", submodules=True
|
|
)
|
|
version(
|
|
"rocm-6.2.0",
|
|
tag="rocm-6.2.0",
|
|
commit="f0bd9126a5456eb9e511d13261af262d17d9b61b",
|
|
submodules=True,
|
|
)
|
|
version(
|
|
"1.11.0", tag="v1.11.0", commit="77d52814e9050004cfb11d7917e155b00ab861b1", submodules=True
|
|
)
|
|
version(
|
|
"1.10.0", tag="v1.10.0", commit="9de3a6b0b4243bf8ec10164babdd99f64dbc65f2", submodules=True
|
|
)
|
|
version(
|
|
"1.9.0", tag="v1.9.0", commit="9eafb2360296277103d2ee706fb5f90b12722668", submodules=True
|
|
)
|
|
version(
|
|
"1.8.0", tag="v1.8.0", commit="7c73d981258cc3a29477756a95c1f90c5f8897dd", submodules=True
|
|
)
|
|
version("1.7.4", commit="12001d9633328f9f56210c7ebffce065bff06310", submodules=True)
|
|
version("1.7.3", commit="2ebfe3fc30f977559142509edc4ea190c975992a", submodules=True)
|
|
version("1.7.2", commit="a41a5c155e0d3780de4c83a76f28d7c8ffa6414f", submodules=True)
|
|
version("1.7.1", commit="67f7471253b8e031e476d80d2bc00e569285c1bf", submodules=True)
|
|
version("1.7.0", commit="2a387f909935d06c6a4874a5b11f38fb8521800e", submodules=True)
|
|
with default_args(deprecated=True):
|
|
version("1.6.0", commit="15e6e6d979fcd5f549d952862400f292ec735b8c", submodules=True)
|
|
version("1.5.0", commit="2718596e5a6808a9278c3f6c8fddfaf977d3bcb6", submodules=True)
|
|
version("1.4.0", commit="23fb3946c7f4c0702b1b168e1d78b8b62597e3f1", submodules=True)
|
|
version("1.3.1", commit="641225f88304909fd2ca5407aec062d0fdf0ed8b", submodules=True)
|
|
version("1.3.0", commit="4dd144a32c8b83c44e132ef53f2b44fe4b4d5569", submodules=True)
|
|
version("1.2.0", commit="f82845388aab108ed1d1fc404f433a0def391bb3", submodules=True)
|
|
|
|
depends_on("c", type="build") # generated
|
|
depends_on("cxx", type="build") # generated
|
|
depends_on("fortran", type="build") # generated
|
|
|
|
variant(
|
|
"rocm",
|
|
default=True,
|
|
description="Enable ROCm API, kernel tracing, and GPU HW counters support",
|
|
)
|
|
variant("strip", default=False, description="Faster binary instrumentation, worse debugging")
|
|
variant(
|
|
"python", default=False, description="Enable support for Python function profiling and API"
|
|
)
|
|
variant("papi", default=True, description="Enable HW counters support via PAPI")
|
|
variant("ompt", default=True, description="Enable OpenMP Tools support")
|
|
variant(
|
|
"tau",
|
|
default=False,
|
|
description="Enable support for using TAU markers in omnitrace instrumentation",
|
|
)
|
|
variant(
|
|
"caliper",
|
|
default=False,
|
|
description="Enable support for using Caliper markers in omnitrace instrumentation",
|
|
)
|
|
variant(
|
|
"perfetto_tools",
|
|
default=False,
|
|
description="Install perfetto tools (e.g. traced, perfetto)",
|
|
)
|
|
variant(
|
|
"mpi",
|
|
default=False,
|
|
description=(
|
|
"Enable intercepting MPI functions and aggregating output during finalization "
|
|
"(requires target application to use same MPI installation)"
|
|
),
|
|
)
|
|
variant(
|
|
"mpi_headers",
|
|
default=True,
|
|
description=(
|
|
"Enable intercepting MPI functions but w/o support for aggregating output "
|
|
"(target application can use any MPI installation)"
|
|
),
|
|
)
|
|
|
|
extends("python", when="+python")
|
|
|
|
# hard dependencies
|
|
depends_on("cmake@3.16:", type="build")
|
|
depends_on("dyninst@11.0.1:", type=("build", "run"))
|
|
depends_on("libunwind", type=("build", "run"))
|
|
|
|
# soft dependencies
|
|
depends_on("hip", when="+rocm")
|
|
depends_on("rocm-smi-lib", when="+rocm")
|
|
depends_on("roctracer-dev", when="+rocm")
|
|
depends_on("rocprofiler-dev", when="@1.3.0: +rocm")
|
|
depends_on("hip@5", when="@1:1.10 +rocm")
|
|
depends_on("rocm-smi-lib@5", when="@1:1.10 +rocm")
|
|
depends_on("roctracer-dev@5", when="@1:1.10 +rocm")
|
|
depends_on("rocprofiler-dev@5", when="@1.3.0:1.10 +rocm")
|
|
|
|
for ver in ["6.2.0", "6.2.1", "6.2.4"]:
|
|
depends_on(f"rocm-smi-lib@{ver}", when=f"@rocm-{ver} +rocm")
|
|
depends_on(f"hip@{ver}", when=f"@rocm-{ver} +rocm")
|
|
depends_on(f"roctracer-dev@{ver}", when=f"@rocm-{ver} +rocm")
|
|
depends_on(f"rocprofiler-dev@{ver}", when=f"@rocm-{ver} +rocm")
|
|
|
|
depends_on("papi+shared", when="+papi")
|
|
depends_on("mpi", when="+mpi")
|
|
depends_on("tau", when="+tau")
|
|
depends_on("caliper", when="+caliper")
|
|
depends_on("python@3:", when="+python", type=("build", "run"))
|
|
depends_on("dyninst@12", when="@1.8:,rocm-6.2:0 +rocm")
|
|
depends_on("m4", when="@1.8:,rocm-6.2:0 +rocm")
|
|
depends_on("texinfo", when="@1.8:,rocm-6.2:0 +rocm")
|
|
depends_on("libunwind", when="@1.8:,rocm-6.2:0 +rocm")
|
|
|
|
def cmake_args(self):
|
|
spec = self.spec
|
|
|
|
args = [
|
|
self.define("SPACK_BUILD", True),
|
|
self.define("OMNITRACE_BUILD_PAPI", False),
|
|
self.define("OMNITRACE_BUILD_PYTHON", True),
|
|
self.define("OMNITRACE_BUILD_DYNINST", False),
|
|
self.define("OMNITRACE_BUILD_LIBUNWIND", False),
|
|
self.define("OMNITRACE_BUILD_STATIC_LIBGCC", False),
|
|
self.define("OMNITRACE_BUILD_STATIC_LIBSTDCXX", False),
|
|
self.define_from_variant("OMNITRACE_BUILD_LTO", "ipo"),
|
|
self.define_from_variant("OMNITRACE_USE_HIP", "rocm"),
|
|
self.define_from_variant("OMNITRACE_USE_MPI", "mpi"),
|
|
self.define_from_variant("OMNITRACE_USE_OMPT", "ompt"),
|
|
self.define_from_variant("OMNITRACE_USE_PAPI", "papi"),
|
|
self.define_from_variant("OMNITRACE_USE_RCCL", "rocm"),
|
|
self.define_from_variant("OMNITRACE_USE_ROCM_SMI", "rocm"),
|
|
self.define_from_variant("OMNITRACE_USE_ROCTRACER", "rocm"),
|
|
self.define_from_variant("OMNITRACE_USE_ROCPROFILER", "rocm"),
|
|
self.define_from_variant("OMNITRACE_USE_PYTHON", "python"),
|
|
self.define_from_variant("OMNITRACE_USE_MPI_HEADERS", "mpi_headers"),
|
|
self.define_from_variant("OMNITRACE_STRIP_LIBRARIES", "strip"),
|
|
self.define_from_variant("OMNITRACE_INSTALL_PERFETTO_TOOLS", "perfetto_tools"),
|
|
# timemory arguments
|
|
self.define("TIMEMORY_UNITY_BUILD", False),
|
|
self.define("TIMEMORY_BUILD_CALIPER", False),
|
|
self.define_from_variant("TIMEMORY_USE_TAU", "tau"),
|
|
self.define_from_variant("TIMEMORY_USE_CALIPER", "caliper"),
|
|
]
|
|
|
|
if "+tau" in spec:
|
|
tau_root = spec["tau"].prefix
|
|
args.append(self.define("TAU_ROOT_DIR", tau_root))
|
|
|
|
if "+mpi" in spec:
|
|
args.append(self.define("MPI_C_COMPILER", spec["mpi"].mpicc))
|
|
args.append(self.define("MPI_CXX_COMPILER", spec["mpi"].mpicxx))
|
|
|
|
if spec.satisfies("@1.8:,rocm-6.2:0"):
|
|
args.append(self.define("dl_LIBRARY", "dl"))
|
|
args.append(
|
|
self.define("libunwind_INCLUDE_DIR", self.spec["libunwind"].prefix.include)
|
|
)
|
|
return args
|
|
|
|
def flag_handler(self, name, flags):
|
|
if self.spec.satisfies("@1.8:,rocm-6.2:0"):
|
|
if name == "ldflags":
|
|
flags.append("-lintl")
|
|
return (flags, None, None)
|
|
|
|
def setup_build_environment(self, env):
|
|
if "+tau" in self.spec:
|
|
import glob
|
|
|
|
# below is how TAU_MAKEFILE is set in packages/tau/package.py
|
|
pattern = join_path(self.spec["tau"].prefix.lib, "Makefile.*")
|
|
files = glob.glob(pattern)
|
|
if files:
|
|
env.set("TAU_MAKEFILE", files[0])
|