From fc19bcbedef95671a7f92f0a2009d5778652ae0c Mon Sep 17 00:00:00 2001
From: Emanuele De Rubeis <ederubei@login01.leonardo.local>
Date: Sat, 30 Dec 2023 19:33:22 +0100
Subject: [PATCH] Full GPU support

---
 Makefile | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)
 mode change 100644 => 100755 Makefile

diff --git a/Makefile b/Makefile
old mode 100644
new mode 100755
index afdd14c..f65a80e
--- a/Makefile
+++ b/Makefile
@@ -56,9 +56,9 @@ OPT += -DPHASE_ON
 #OPT += -DNORMALIZE_UVW
 
 # Gridding kernel: GAUSS, GAUSS_HI_PRECISION, KAISERBESSEL
-#OPT += -DGAUSS_HI_PRECISION
+OPT += -DGAUSS_HI_PRECISION
 
-OPT += -DGAUSS
+#OPT += -DGAUSS
 
 #OPT += -DKAISERBESSEL
 
@@ -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)))
 FLAGS=$(OPTIMIZE)
@@ -311,7 +318,11 @@ ifeq (CUDACC,$(findstring CUDACC,$(OPT)))
 EXEC_EXT := $(EXEC_EXT)_acc-fft
 LINKER=$(MPIC++)
 FLAGS=$(OPTIMIZE)
+ifeq (NCCL_REDUCE,$(findstring NCCL_REDUCE,$(OPT)))
+LIBS=$(NVLIB_2) $(NVLIB_3)
+else
 LIBS=$(NVLIB_2)
+endif
 $(OBJ_ACC_CUFFTMP): $(DEPS_ACC_CUFFTMP)
 	$(NVCC) $(OPT_NVCC) $(OPT) -c $^ $(LIBS)
 OBJ += $(OBJ_ACC_CUFFTMP)
-- 
GitLab