hlrs-software-stack/spack
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

kineto: add new package (#24319)

Browse source
This commit is contained in:
Adam J. Stewart 2021-06-16 08:43:47 -05:00 committed by GitHub
parent 38088dd898
commit 72c6fc2fda
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 28 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
var/spack/repos/builtin/packages/kineto/package.py Normal file
View file

@ -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'