Combine Kokkos CPU and GPU builds
This commit is contained in:
parent
584e1b208b
commit
ce5fde973b
28
Kokkos.make
Normal file
28
Kokkos.make
Normal 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
|
||||||
|
|
||||||
@ -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
|
|
||||||
|
|
||||||
@ -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
|
|
||||||
|
|
||||||
Loading…
Reference in New Issue
Block a user