Sylvain Jeaugey
ffc55560a1
Add tree algorithms for allreduce to improve performance at scale.
Add ncclCommAbort() and ncclCommGetAsyncError() to properly handle
network errors and be permit recover.
Detect initial CPU affinity and no longer escape it.
[ROCm/rccl commit: 1450d42675]
41 řádky
1.0 KiB
Makefile
41 řádky
1.0 KiB
Makefile
#
|
|
# Copyright (c) 2015-2018, NVIDIA CORPORATION. All rights reserved.
|
|
#
|
|
# See LICENSE.txt for license information
|
|
#
|
|
|
|
include ../../makefiles/common.mk
|
|
include ../../makefiles/version.mk
|
|
BUILDDIR ?= $(abspath ../../build)
|
|
TXZPREPDIR := $(BUILDDIR)/srctxz
|
|
PKGDIR := $(BUILDDIR)/pkg/srctxz/
|
|
|
|
TXZGEN_IN := $(wildcard *.in)
|
|
TXZGEN := $(TXZGEN_IN:.in=)
|
|
TXZTARGETS := $(patsubst %, $(TXZPREPDIR)/%, $(TXZGEN))
|
|
|
|
PKG_REVISION ?= 3
|
|
PKG_ARCH := $(shell uname -m)
|
|
|
|
prep: $(TXZTARGETS)
|
|
|
|
build: prep
|
|
$(MAKE) -C ../../src clean
|
|
@printf "Building source tar.xz package\n"
|
|
(cd $(BUILDDIR); bash srctxz/create_srctxz.sh)
|
|
mkdir -p $(PKGDIR)
|
|
mv $(BUILDDIR)/../../nccl-src*.txz $(PKGDIR)
|
|
|
|
clean:
|
|
rm -Rf $(TXZPREPDIR) $(PKGDIR)
|
|
|
|
$(TXZPREPDIR)/% : %.in
|
|
@printf "Generating %-35s > %s\n" $< $@
|
|
mkdir -p $(TXZPREPDIR)
|
|
sed -e "s/\$${nccl:Major}/$(NCCL_MAJOR)/g" \
|
|
-e "s/\$${nccl:Minor}/$(NCCL_MINOR)/g" \
|
|
-e "s/\$${nccl:Patch}/$(NCCL_PATCH)/g" \
|
|
-e "s/\$${nccl:Suffix}/$(NCCL_SUFFIX)/g" \
|
|
-e "s/\$${pkg:Revision}/$(PKG_REVISION)/g" \
|
|
$< > $@
|