Combine Kokkos CPU and GPU builds

This commit is contained in:
Tom Deakin 2017-02-23 15:31:38 +00:00
parent 584e1b208b
commit ce5fde973b
3 changed files with 28 additions and 30 deletions

28
Kokkos.make Normal file
View File

@ -0,0 +1,28 @@
default: kokkos-stream
include $(KOKKOS_PATH)/Makefile.kokkos
ifndef TARGET
$(info No target defined. Specify CPU or GPU. Defaulting to CPU)
TARGET=CPU
endif
ifeq ($(TARGET), CPU)
COMPILER = $(CXX)
TARGET_DEF = -DKOKKOS_TARGET_CPU
else ifeq ($(TARGET), GPU)
COMPILER = $(NVCC_WRAPPER)
TARGET_DEF =
endif
kokkos-stream: main.o KOKKOSStream.o
$(CXX) $(KOKKOS_LDFLAGS) $^ $(KOKKOS_LIBS) -o $@ -DKOKKOS $(TARGET_DEF) -O3
%.o:%.cpp $(KOKKOS_CPP_DEPENDS)
$(COMPILER) $(KOKKOS_CPPFLAGS) $(KOKKOS_CXXFLAGS) -c $< -DKOKKOS $(TARGET_DEF) -O3
.PHONY: clean
clean:
rm -f main.o KOKKOSStream.o

View File

@ -1,15 +0,0 @@
default: kokkos-cpu-stream
include $(KOKKOS_PATH)/Makefile.kokkos
kokkos-cpu-stream: main.o KOKKOSStream.o
$(CXX) $(KOKKOS_LDFLAGS) $^ $(KOKKOS_LIBS) -o $@ -DKOKKOS -DKOKKOS_TARGET_CPU -O3
%.o:%.cpp $(KOKKOS_CPP_DEPENDS)
$(CXX) $(KOKKOS_CPPFLAGS) $(KOKKOS_CXXFLAGS) -c $< -DKOKKOS -DKOKKOS_TARGET_CPU -O3
.PHONY: clean
clean:
rm -f main.o KOKKOSStream.o

View File

@ -1,15 +0,0 @@
default: kokkos-gpu-stream
include $(KOKKOS_PATH)/Makefile.kokkos
kokkos-gpu-stream: main.o KOKKOSStream.o
$(CXX) $(KOKKOS_LDFLAGS) $^ $(KOKKOS_LIBS) -o $@ -DKOKKOS -O3
%.o:%.cpp $(KOKKOS_CPP_DEPENDS)
$(NVCC_WRAPPER) $(KOKKOS_CPPFLAGS) $(KOKKOS_CXXFLAGS) -c $< -DKOKKOS -O3
.PHONY: clean
clean:
rm -f main.o KOKKOSStream.o