Skip to content
Snippets Groups Projects
Commit fc19bcbe authored by Emanuele De Rubeis's avatar Emanuele De Rubeis
Browse files

Full GPU support

parent f71d835e
No related branches found
No related tags found
No related merge requests found
Makefile 100644 → 100755
...@@ -56,9 +56,9 @@ OPT += -DPHASE_ON ...@@ -56,9 +56,9 @@ OPT += -DPHASE_ON
#OPT += -DNORMALIZE_UVW #OPT += -DNORMALIZE_UVW
# Gridding kernel: GAUSS, GAUSS_HI_PRECISION, KAISERBESSEL # Gridding kernel: GAUSS, GAUSS_HI_PRECISION, KAISERBESSEL
#OPT += -DGAUSS_HI_PRECISION OPT += -DGAUSS_HI_PRECISION
OPT += -DGAUSS #OPT += -DGAUSS
#OPT += -DKAISERBESSEL #OPT += -DKAISERBESSEL
...@@ -96,6 +96,13 @@ OPT += -DGAUSS ...@@ -96,6 +96,13 @@ OPT += -DGAUSS
# ======================================================== # ========================================================
# ========================================================
#FULL GPU SUPPORT!!!
OPT += -DFULL_NVIDIA
ifeq (FULL_NVIDIA,$(findstring FULL_NVIDIA,$(OPT)))
OPT += -DCUDACC -DNCCL_REDUCE -DCUFFTMP
endif
# =======================================================
ifeq (USE_OMP,$(findstring USE_OMP,$(OPT))) ifeq (USE_OMP,$(findstring USE_OMP,$(OPT)))
FLAGS=$(OPTIMIZE) FLAGS=$(OPTIMIZE)
...@@ -311,7 +318,11 @@ ifeq (CUDACC,$(findstring CUDACC,$(OPT))) ...@@ -311,7 +318,11 @@ ifeq (CUDACC,$(findstring CUDACC,$(OPT)))
EXEC_EXT := $(EXEC_EXT)_acc-fft EXEC_EXT := $(EXEC_EXT)_acc-fft
LINKER=$(MPIC++) LINKER=$(MPIC++)
FLAGS=$(OPTIMIZE) FLAGS=$(OPTIMIZE)
ifeq (NCCL_REDUCE,$(findstring NCCL_REDUCE,$(OPT)))
LIBS=$(NVLIB_2) $(NVLIB_3)
else
LIBS=$(NVLIB_2) LIBS=$(NVLIB_2)
endif
$(OBJ_ACC_CUFFTMP): $(DEPS_ACC_CUFFTMP) $(OBJ_ACC_CUFFTMP): $(DEPS_ACC_CUFFTMP)
$(NVCC) $(OPT_NVCC) $(OPT) -c $^ $(LIBS) $(NVCC) $(OPT_NVCC) $(OPT) -c $^ $(LIBS)
OBJ += $(OBJ_ACC_CUFFTMP) OBJ += $(OBJ_ACC_CUFFTMP)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment