diff options
| author | Adam J. Stewart <ajstewart426@gmail.com> | 2021-06-16 08:43:47 -0500 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-06-16 07:43:47 -0600 |
| commit | 72c6fc2fdaf1d516a9692f6ccf8ac5b8a0c817fb (patch) | |
| tree | a20ffad3d540775fbc835fead2458c7a094b601f | |
| parent | 38088dd89860e094723c1d0fee92e43afceeef12 (diff) | |
| download | spack-72c6fc2fdaf1d516a9692f6ccf8ac5b8a0c817fb.tar.gz spack-72c6fc2fdaf1d516a9692f6ccf8ac5b8a0c817fb.tar.bz2 spack-72c6fc2fdaf1d516a9692f6ccf8ac5b8a0c817fb.tar.xz spack-72c6fc2fdaf1d516a9692f6ccf8ac5b8a0c817fb.zip | |
kineto: add new package (#24319)
| -rw-r--r-- | var/spack/repos/builtin/packages/kineto/package.py | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/kineto/package.py b/var/spack/repos/builtin/packages/kineto/package.py new file mode 100644 index 0000000000..407b4deac7 --- /dev/null +++ b/var/spack/repos/builtin/packages/kineto/package.py @@ -0,0 +1,28 @@ +# Copyright 2013-2021 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 import * + + +class Kineto(CMakePackage): + """A CPU+GPU Profiling library that provides access to timeline traces + and hardware performance counters.""" + + homepage = "https://github.com/pytorch/kineto" + git = "https://github.com/pytorch/kineto.git" + + version('master', branch='master', submodules=True) + version('2021-05-12', commit='a631215ac294805d5360e0ecceceb34de6557ba8', submodules=True) # py-torch@1.9 + version('2021-03-16', commit='ce98f8b95b2ee5ffed257ca90090cd8adcf15b53', submodules=True) # py-torch@1.8.1 + version('2021-02-04', commit='258d9a471f8d3a50a0f52b85c3fe0902f65489df', submodules=True) # py-torch@1.8.0 + + root_cmakelists_dir = 'libkineto' + + depends_on('cmake@3.5:', type='build') + depends_on('ninja', type='build') + depends_on('python', type='build') + depends_on('cuda') + + generator = 'Ninja' |