diff --git a/Kokkos.make b/Kokkos.make index a09f762..e46236e 100644 --- a/Kokkos.make +++ b/Kokkos.make @@ -36,15 +36,18 @@ ifndef COMPILER define compiler_help Set COMPILER to change flags (defaulting to GNU). Available compilers are: - GNU INTEL + GNU INTEL CRAY PGI ARMCLANG endef $(info $(compiler_help)) COMPILER=GNU endif +COMPILER_ARMCLANG = armclang++ COMPILER_GNU = g++ COMPILER_INTEL = icpc -qopt-streaming-stores=always +COMPILER_CRAY = CC +COMPILER_PGI = pgc++ CXX = $(COMPILER_$(COMPILER)) ifndef TARGET @@ -67,6 +70,14 @@ CXXFLAGS = -O3 LINKFLAGS = # empty for now + +ifeq ($(COMPILER), GNU) +ifeq ($(DEVICE), OpenMP) +CXXFLAGS += -fopenmp +LINKFLAGS += -fopenmp +endif +endif + include $(KOKKOS_PATH)/Makefile.kokkos kokkos-stream: $(OBJ) $(KOKKOS_LINK_DEPENDS) diff --git a/OpenMP.make b/OpenMP.make index d9e331a..5ead9ce 100644 --- a/OpenMP.make +++ b/OpenMP.make @@ -3,7 +3,7 @@ ifndef COMPILER define compiler_help Set COMPILER to change flags (defaulting to GNU). Available compilers are: - CLANG CRAY GNU INTEL XL PGI NEC + CLANG CRAY GNU INTEL XL PGI NEC ARMCLANG endef $(info $(compiler_help)) @@ -21,6 +21,7 @@ $(info $(target_help)) TARGET=CPU endif +COMPILER_ARMCLANG = armclang++ COMPILER_GNU = g++ COMPILER_INTEL = icpc COMPILER_CRAY = CC