From 9a71a7646130017c767a3a85c642fc2b097d824d Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Fri, 20 Mar 2020 05:08:54 -0400 Subject: py-tfdlpack: add new package (#15583) --- .../repos/builtin/packages/py-tfdlpack/package.py | 45 ++++++++++++++++++++++ 1 file changed, 45 insertions(+) create mode 100644 var/spack/repos/builtin/packages/py-tfdlpack/package.py diff --git a/var/spack/repos/builtin/packages/py-tfdlpack/package.py b/var/spack/repos/builtin/packages/py-tfdlpack/package.py new file mode 100644 index 0000000000..678cc2d6db --- /dev/null +++ b/var/spack/repos/builtin/packages/py-tfdlpack/package.py @@ -0,0 +1,45 @@ +# Copyright 2013-2020 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) + + +class PyTfdlpack(CMakePackage): + """Tensorflow plugin for DLPack.""" + + homepage = "https://github.com/VoVAllen/tf-dlpack" + git = "https://github.com/VoVAllen/tf-dlpack.git" + + maintainers = ['adamjstewart'] + + version('master', branch='master', submodules=True) + version('0.1.1', tag='v0.1.1', submodules=True) + + variant('cuda', default=True, description='Build with CUDA support') + + depends_on('cmake@3.5:', type='build') + depends_on('cuda', when='+cuda') + + # Python dependencies + extends('python') + depends_on('py-setuptools', type='build') + depends_on('py-tensorflow', type=('build', 'run')) + + def cmake_args(self): + args = ['-DPYTHON_EXECUTABLE=' + self.spec['python'].command.path] + + if '+cuda' in self.spec: + args.append('-DUSE_CUDA=ON') + else: + args.append('-DUSE_CUDA=OFF') + + return args + + def install(self, spec, prefix): + with working_dir('python'): + setup_py('install', '--prefix=' + prefix, + '--single-version-externally-managed', '--root=/') + + def setup_run_environment(self, env): + # Prevent TensorFlow from taking over the whole GPU + env.set('TF_FORCE_GPU_ALLOW_GROWTH', 'true') -- cgit v1.2.3-70-g09d2