Makefile

GPU=AMD
GPU=NVIDIA

OBJ_blas = blas.o
OBJ_fftw = fftw.o
OBJ_tensor_core = tensor_core.o


ifeq ($(GPU), AMD)
  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 = -lm -lgomp
  LDFLAGS_blas =  -lcublas
  LDFLAGS_fftw =  -lcufft
  INCLUDE= -I/usr/lib/x86_64-linux-gnu/openmpi/include/
  CFLAGS = ${INCLUDE}  -arch sm_70 --compile -O3 -pg -Xcompiler -fopenmp -DCUDA
  CUDAFLAGS = -arch sm_70  --Werror cross-execution-space-call --Wno-deprecated-gpu-targets
else
  unknown_HW:
endif

all: blas fftw

blas: ${OBJ_blas}
	${CC} -o blas ${OBJ_blas} ${LDFLAGS} ${LDFLAGS_blas} ${CUDAFLAGS}

tensor_core: ${OBJ_tensor_core}
	${CC} -o tensor_core ${OBJ_tensor_core} ${LDFLAGS} ${LDFLAGS_blas} ${CUDAFLAGS}

fftw: ${OBJ_fftw}
	${CC} -o fftw ${OBJ_fftw} ${LDFLAGS} ${LDFLAGS_fftw} ${CUDAFLAGS}

%.o: %.c ${HEADER}
	${CC} ${CFLAGS} -c $<

clean:
	rm *.o

unknown_HW:
	@echo "hardware not detected"