aparler/PyNorch
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix send to other devices and compile on cloud

Browse source
This commit is contained in:
lucasdelimanogueira 2024-05-29 17:48:37 -03:00
parent 446b382333
commit 38a860eba1
4 changed files with 22 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

2
build/Makefile
View file

@ -13,7 +13,7 @@ CUDAFLAGS = -arch=sm_75
CUDALIBS = -I/usr/local/cuda/include -L/usr/local/cuda/lib64 -lcudart -lcuda
# MPI flags and libraries
DISTRIBUTEDLIBS = -lmpi_cxx -lnccl
DISTRIBUTEDLIBS = -lmpi_cxx -lnccl -lmpi
# Directories
SRCDIR = ../norch/csrc