From 3af971165a4e6a682fb40fcef0e9c68e5c905d8c Mon Sep 17 00:00:00 2001 From: Brian Van Essen Date: Fri, 24 Mar 2017 17:53:16 -0700 Subject: Added a package for Nvida NCCL (#3552) --- var/spack/repos/builtin/packages/nccl/package.py | 41 ++++++++++++++++++++++++ 1 file changed, 41 insertions(+) create mode 100644 var/spack/repos/builtin/packages/nccl/package.py diff --git a/var/spack/repos/builtin/packages/nccl/package.py b/var/spack/repos/builtin/packages/nccl/package.py new file mode 100644 index 0000000000..b68d3429d1 --- /dev/null +++ b/var/spack/repos/builtin/packages/nccl/package.py @@ -0,0 +1,41 @@ +############################################################################## +# Copyright (c) 2013-2016, Lawrence Livermore National Security, LLC. +# Produced at the Lawrence Livermore National Laboratory. +# +# This file is part of Spack. +# Created by Todd Gamblin, tgamblin@llnl.gov, All rights reserved. +# LLNL-CODE-647188 +# +# For details, see https://github.com/llnl/spack +# Please also see the LICENSE file for our notice and the LGPL. +# +# This program is free software; you can redistribute it and/or modify +# it under the terms of the GNU Lesser General Public License (as +# published by the Free Software Foundation) version 2.1, February 1999. +# +# This program is distributed in the hope that it will be useful, but +# WITHOUT ANY WARRANTY; without even the IMPLIED WARRANTY OF +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the terms and +# conditions of the GNU Lesser General Public License for more details. +# +# You should have received a copy of the GNU Lesser General Public +# License along with this program; if not, write to the Free Software +# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA +############################################################################## +from spack import * + + +class Nccl(MakefilePackage): + """Optimized primitives for collective multi-GPU communication.""" + + homepage = "https://github.com/NVIDIA/nccl" + url = "https://github.com/NVIDIA/nccl/archive/v1.3.4-1.tar.gz" + + version('1.3.4-1', '5b9ce7fbdce0fde68e0f66318e6ff422') + version('1.3.0-1', 'f6fb1d56913a7d212ca0c300e76f01fb') + + depends_on('cuda') + + @property + def install_targets(self): + return ['PREFIX={0}'.format(self.prefix), 'install'] -- cgit v1.2.3-70-g09d2