NR_TASKLETS ?= 16
BL ?= 8
NR_DPUS ?= 0
NR_RANKS ?= 0
TRANSFER ?= PUSH
DPU_BINARY ?= '"bin/dpu_code"'
NUMA ?= 0

COMMON_INCLUDES := support
HOST_SOURCES := $(wildcard host/*.c)

COMMON_FLAGS := -Wall -Wextra -g -I${COMMON_INCLUDES}
HOST_FLAGS := ${COMMON_FLAGS} -D_POSIX_C_SOURCE=200809L -std=c11 -O3 `dpu-pkg-config --cflags --libs dpu` -DNR_TASKLETS=${NR_TASKLETS} -DNR_DPUS=${NR_DPUS} -DNR_RANKS=${NR_RANKS} -DBL=${BL} -D${TRANSFER} -DDPU_BINARY=${DPU_BINARY} -DNUMA=${NUMA}
DPU_FLAGS := ${COMMON_FLAGS} -O2 -DNR_TASKLETS=${NR_TASKLETS} -DBL=${BL} -D${TRANSFER}

QUIET = @

ifdef verbose
	QUIET =
endif

ifeq (${NUMA}, 1)
	HOST_FLAGS += -lnuma
endif

all: bin/host_code bin/dpu_code bin/dpu_size

bin:
	${QUIET}mkdir -p bin

bin/host_code: ${HOST_SOURCES} ${COMMON_INCLUDES} bin
	${QUIET}${CC} -o $@ ${HOST_SOURCES} ${HOST_FLAGS}

bin/dpu_code: dpu/task.c ${COMMON_INCLUDES} bin
	${QUIET}dpu-upmem-dpurte-clang ${DPU_FLAGS} -o $@ dpu/task.c

bin/dpu_size: dpu/size.c ${COMMON_INCLUDES} bin
	${QUIET}dpu-upmem-dpurte-clang ${DPU_FLAGS} -o $@ dpu/size.c

clean:
	${QUIET}rm -rf bin

test: all
	${QUIET}bin/host_code

.PHONY: all clean test