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