2018-09-24 16:06:59 -07:00
|
|
|
#!/bin/bash
|
|
|
|
|
#
|
2019-03-14 19:39:20 -07:00
|
|
|
# Copyright (c) 2017-2019, NVIDIA CORPORATION. All rights reserved.
|
2018-09-24 16:06:59 -07:00
|
|
|
#
|
|
|
|
|
# See LICENSE.txt for license information
|
|
|
|
|
#
|
|
|
|
|
|
|
|
|
|
# To run from $BUILDDIR/
|
|
|
|
|
|
|
|
|
|
BUILDDIR=`basename $PWD`
|
|
|
|
|
|
|
|
|
|
cd ..
|
|
|
|
|
NCCL_MAJOR=${nccl:Major}
|
|
|
|
|
NCCL_MINOR=${nccl:Minor}
|
|
|
|
|
NCCL_PATCH=${nccl:Patch}
|
|
|
|
|
NCCL_SUFFIX=${nccl:Suffix}
|
|
|
|
|
CUDA_MAJOR=${cuda:Major}
|
|
|
|
|
CUDA_MINOR=${cuda:Minor}
|
|
|
|
|
PKG_REVISION=${pkg:Revision}
|
|
|
|
|
PKG_ARCH=${pkg:Arch}
|
|
|
|
|
|
|
|
|
|
NCCLNAME="nccl_${NCCL_MAJOR}.${NCCL_MINOR}.${NCCL_PATCH}${NCCL_SUFFIX}-${PKG_REVISION}+cuda${CUDA_MAJOR}.${CUDA_MINOR}_${PKG_ARCH}"
|
|
|
|
|
|
|
|
|
|
tar --transform "s/^$BUILDDIR/$NCCLNAME/" -Jcf $NCCLNAME.txz --owner=0 --group=0 $BUILDDIR/include $BUILDDIR/lib $BUILDDIR/*.txt
|
