diff options
author | Hervé Yviquel <hyviquel@gmail.com> | 2021-05-25 14:34:28 -0300 |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-05-25 10:34:28 -0700 |
commit | e22da8df057c607ddc586956caa60a76ff25574d (patch) | |
tree | 704beff34f0ed4eaf347af2b4a408e9a83d2c1e3 /var | |
parent | c5389c430b0923b465cb80490ad2a89b1fc5216f (diff) | |
download | spack-e22da8df057c607ddc586956caa60a76ff25574d.tar.gz spack-e22da8df057c607ddc586956caa60a76ff25574d.tar.bz2 spack-e22da8df057c607ddc586956caa60a76ff25574d.tar.xz spack-e22da8df057c607ddc586956caa60a76ff25574d.zip |
extrae package: add variant for cuda and cupti (#23522)
Diffstat (limited to 'var')
-rw-r--r-- | var/spack/repos/builtin/packages/extrae/package.py | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/extrae/package.py b/var/spack/repos/builtin/packages/extrae/package.py index 07a7d52150..d4531876e0 100644 --- a/var/spack/repos/builtin/packages/extrae/package.py +++ b/var/spack/repos/builtin/packages/extrae/package.py @@ -67,6 +67,13 @@ class Extrae(AutotoolsPackage): variant('papi', default=True, description="Use PAPI to collect performance counters") depends_on('papi', when='+papi') + variant('cuda', default=False, description="Enable support for tracing CUDA") + depends_on('cuda', when='+cuda') + + variant('cupti', default=False, description='Enable CUPTI support') + depends_on('cuda', when='+cupti') + conflicts('+cupti', when='~cuda', msg='CUPTI requires CUDA') + def configure_args(self): spec = self.spec args = ["--with-mpi=%s" % spec['mpi'].prefix, @@ -85,6 +92,19 @@ class Extrae(AutotoolsPackage): if '+dyninst' in self.spec else ["--without-dyninst"]) + args += (["--with-cuda=%s" % spec['cuda'].prefix] + if '+cuda' in self.spec else + ["--without-cuda"]) + + if '+cupti' in self.spec: + cupti_h = find_headers('cupti', spec['cuda'].prefix, + recursive=True) + cupti_dir = os.path.dirname(os.path.dirname(cupti_h[0])) + + args += (["--with-cupti=%s" % cupti_dir] + if '+cupti' in self.spec else + ["--without-cupti"]) + if spec.satisfies("^dyninst@9.3.0:"): make.add_default_arg("CXXFLAGS=%s" % self.compiler.cxx11_flag) args.append("CXXFLAGS=%s" % self.compiler.cxx11_flag) |