2018-11-13 10:37:20 -08:00
|
|
|
#
|
2019-03-14 19:39:20 -07:00
|
|
|
# Copyright (c) 2015-2019, NVIDIA CORPORATION. All rights reserved.
|
2018-11-13 10:37:20 -08:00
|
|
|
#
|
|
|
|
|
# See LICENSE.txt for license information
|
|
|
|
|
#
|
|
|
|
|
NCCL_HOME:=../../build/
|
|
|
|
|
CUDA_HOME:=/usr/local/cuda
|
|
|
|
|
INC:= -I$(NCCL_HOME)/include -I$(CUDA_HOME)/include
|
|
|
|
|
PLUGIN_SO:=libnccl-net.so
|
|
|
|
|
|
|
|
|
|
default: $(PLUGIN_SO)
|
|
|
|
|
|
|
|
|
|
$(PLUGIN_SO): plugin.c
|
|
|
|
|
$(CC) $(INC) -fPIC -shared -o $@ -Wl,-soname,$(PLUGIN_SO) $^
|
|
|
|
|
|
|
|
|
|
clean:
|
|
|
|
|
rm -f $(PLUGIN_SO)
|
