Verified Commit 39e09f0d authored by Kelleher, Michael's avatar Kelleher, Michael
Browse files

Fix nvidia arch for AFW

parent 0deb6422
Loading
Loading
Loading
Loading
+2 −4
Original line number Diff line number Diff line
ifdef ACTIVE_GPU
	CC := nvcc -ccbin CC
	INC_DIRS := $(MPICH_DIR)/include
	FLAGS := -arch=sm_70 -x cu
	FLAGS := -arch=sm_80 -x cu
	DFLAGS := -DACTIVE_GPU=1
	LIBRARIES := -L$(MPICH_DIR)/lib -lcuda $(PE_MPICH_GTL_DIR_nvidia70) $(PE_MPICH_GTL_LIBS_nvidia70) -lmpi
	# Holding these for GPU aware MPI
	# $(PE_MPICH_GTL_DIR_nvidia70) $(PE_MPICH_GTL_LIBS_nvidia70)
	LIBRARIES := -L$(MPICH_DIR)/lib -lcuda $(PE_MPICH_GTL_DIR_nvidia80) $(PE_MPICH_GTL_LIBS_nvidia80) -lmpi
else
	CC := CC
	INC_DIRS := $(MPICH_DIR)/include