From d3142d28eeadca3f708be2f87f4c855b3f43db99 Mon Sep 17 00:00:00 2001 From: fehrmann <henning.fehrmann@aei.mpg.de> Date: Fri, 29 Jan 2021 10:05:53 +0100 Subject: [PATCH] add fftw compilation --- Makefile | 18 +++++++++++++----- 1 file changed, 13 insertions(+), 5 deletions(-) diff --git a/Makefile b/Makefile index 7ced09e..431692f 100644 --- a/Makefile +++ b/Makefile @@ -2,16 +2,20 @@ GPU=NVIDIA GPU=AMD -OBJ = blas.o +OBJ_blas = blas.o +OBJ_fftw = fftw.o ifeq ($(GPU), AMD) - LDFLAGS = -L/opt/rocm/lib -lhipblas -lrocblas -fopenmp + LDFLAGS = -L/opt/rocm/lib -fopenmp + LDFLAGS_blas = -lhipblas -lrocblas + LDFLAGS_fftw = -lrocfft CFLAGS = -g -Wall -O3 -fopenmp -I/opt/rocm/include -I/opt/rocm/hip/include -DROC CC = hipcc else ifeq ($(GPU), NVIDIA) CC = nvcc - LDFLAGS = -lcublas -lm -lgomp + LDFLAGS = -lm -lgomp + LDFLAGS_blas = -lcublas INCLUDE= -I/usr/lib/x86_64-linux-gnu/openmpi/include/ CFLAGS = ${INCLUDE} --compile -O3 -pg -Xcompiler -fopenmp -DCUDA CUDAFLAGS = --Werror cross-execution-space-call --Wno-deprecated-gpu-targets @@ -19,9 +23,13 @@ else unknown_HW: endif +all: blas fftw -all: ${OBJ} - ${CC} -o blas ${OBJ} ${LDFLAGS} ${CUDAFLAGS} +blas: ${OBJ_blas} + ${CC} -o blas ${OBJ_blas} ${LDFLAGS} ${LDFLAGS_blas} ${CUDAFLAGS} + +fftw: ${OBJ_fftw} + ${CC} -o fftw ${OBJ_fftw} ${LDFLAGS} ${LDFLAGS_fftw} ${CUDAFLAGS} %.o: %.c ${HEADER} ${CC} ${CFLAGS} -c $< -- GitLab