Verified Commit 0f3981e5 authored by Kelleher, Michael's avatar Kelleher, Michael
Browse files

Improve AFW makefile with environment variables 

Also add aliases for platform specific GPU names
parent 522c23c4

makefile

+4 −0
Original line number Diff line number Diff line
@@ -19,11 +19,15 @@ summit_omp: 
afw_gpu:

	make -f makefile.afw ACTIVE_GPU=1 SRCDIR=$(SRCDIR) BUILDDIR=$(BUILDDIR)



afw_cuda:	afw_gpu



afw_omp:

	make -f makefile.afw ACTIVE_OMP=1 SRCDIR=$(SRCDIR) BUILDDIR=$(BUILDDIR)



frontier_gpu:

	make -f makefile.frontier ACTIVE_GPU=1 SRCDIR=$(SRCDIR) BUILDDIR=$(BUILDDIR)



frontier_hip:	frontier_gpu



clean:

	rm -rf $(BUILDDIR)/triton

makefile.afw

+19 −19
Original line number Diff line number Diff line 
ifdef ACTIVE_GPU

	CC := nvcc -ccbin CC

	INC_DIRS := /opt/cray/pe/mpich/8.1.6/ofi/gnu/9.1/include

	INC_DIRS := $(MPICH_DIR)/include

	FLAGS := -arch=sm_70 -x cu

	LIBRARIES := -L$(CUDA_DIR)/targets/ppc64le-linux/lib/stubs -L/opt/cray/pe/mpich/8.1.6/ofi/gnu/9.1/lib -lnvidia-ml

	DFLAGS := -DACTIVE_GPU=1

	LIBRARIES := -L$(MPICH_DIR)/lib -lnvidia-ml -lcuda $(PE_MPICH_GTL_DIR_nvidia70) $(PE_MPICH_GTL_LIBS_nvidia70)

else

	CC := CC

	INC_DIRS := /opt/cray/pe/mpich/8.1.6/ofi/gnu/9.1/include

	INC_DIRS := $(MPICH_DIR)/include

	FLAGS := -Wall -fopenmp

	LIBRARIES := -L/opt/cray/pe/mpich/8.1.6/ofi/gnu/9.1/lib 

	LIBRARIES := -L$(MPICH_DIR)/lib

	DFLAGS := -DACTIVE_OMP=1

endif