both GNU and NVIDIA compilers supported

Author: Maochao Xiao

.gitignore

@@ -32,4 +32,4 @@
 *.app
 
 # build 
-./build
+build/

Makefile

@@ -1,18 +1,32 @@
-# FC = nvfortran
-FC = mpifort
+# Compiler type: GNU or NVIDIA
+COMPILER = NVIDIA
 # Set SINGLE_PRECISION to 1 to use single precision, otherwise use double precision
+SINGLE_PRECISION = 0
+# SmartRedis installation directory
+SMARTREDIS_INSTALL_DIR = /leonardo/home/userexternal/mxiao000/code/smartredis-nvidia/install
+
+# Set module flag based on compiler type
+ifeq ($(COMPILER),GNU)
+  FC = mpifort
+  MODULE_FLAG = -J
+else ifeq ($(COMPILER),NVIDIA)
+  FC = mpifort
+  MODULE_FLAG = -module
+else
+  $(error Invalid COMPILER value. Must be GNU or NVIDIA)
+endif
+
 # For a library dealing with MPI communication and data processing, 
 # -O2 is typically sufficient and recommended for production use.
-SINGLE_PRECISION = 1
-FCFLAGS = -O2 -fPIC -I/scratch/yuningw/reinforcement_learning/04-SmartSOD2D-Reproduce/envs/gpu_smartredis/install/include -cpp
+FCFLAGS = -O2 -fPIC -I$(SMARTREDIS_INSTALL_DIR)/include -cpp
 
 # Add single precision flag if enabled, i.e. make SINGLE_PRECISION=1
 ifeq ($(SINGLE_PRECISION),1)
 FCFLAGS += -D_SINGLE_PRECISION
 endif
 
 # Add SmartRedis library paths and libraries
-LDFLAGS = -L/scratch/yuningw/reinforcement_learning/04-SmartSOD2D-Reproduce/envs/gpu_smartredis/install/lib
+LDFLAGS = -L$(SMARTREDIS_INSTALL_DIR)/lib
 LIBS = -lsmartredis -lsmartredis-fortran
 
 # Directory structure
@@ -21,11 +35,8 @@ OBJ_DIR = $(BUILD_DIR)
 MOD_DIR = $(BUILD_DIR)/include
 LIB_DIR = $(BUILD_DIR)/lib
 
-
-# YW add
 FCFLAGS += -I$(MOD_DIR)
 
-
 # Source files (adjust paths if necessary)
 SRCS = src/precision.f90 src/smartredis_mpi.f90
 OBJS = $(patsubst src/%.f90,$(OBJ_DIR)/%.o,$(SRCS))
@@ -42,7 +53,7 @@ create_dirs:
 	mkdir -p $(LIB_DIR)
 
 $(OBJ_DIR)/%.o: src/%.f90
-	$(FC) $(FCFLAGS) -module $(MOD_DIR) -c $< -o $@
+	$(FC) $(FCFLAGS) $(MODULE_FLAG) $(MOD_DIR) -c $< -o $@
 
 $(LIB_STATIC): $(OBJS)
 	ar rcs $@ $(OBJS)
@@ -51,4 +62,14 @@ $(LIB_SHARED): $(OBJS)
 	$(FC) -shared -o $@ $(OBJS) $(LDFLAGS) $(LIBS)
 
 clean:
-	rm -rf $(BUILD_DIR)
+	rm -rf $(BUILD_DIR)
+
+# Debug targets to print variables
+debug:
+	@echo "COMPILER_TYPE=$(COMPILER_TYPE)"
+	@echo "FC=$(FC)"
+	@echo "FCFLAGS=$(FCFLAGS)"
+
+# Generic target to print any variable, usage: make print-VARIABLE_NAME
+print-%:
+	@echo $*=$($*)

README.md

@@ -43,6 +43,7 @@ The library is functional and in active use. Development is ongoing to implement
 SmartRedis-MPI is developed and maintained by a dedicated team of researchers and developers. We are grateful for the contributions of:
 
 - **Maochao Xiao** ([@soaringxmc](https://github.com/soaringxmc)) - Code implementation and library design
+- **Yuning Wang** ([@fantasy98](https://github.com/Fantasy98)) - Code implementation
 - **Bernat Font** ([@b-fg](https://github.com/b-fg)) - Library design
 - **Marius Kurz** ([@m-kurz](https://github.com/m-kurz)) - Library design
 
@@ -54,7 +55,7 @@ SmartRedis-MPI has benefited from the expertise and support of many individuals
 We are grateful for the valuable discussions with:
 - Francisco Alcántara Ávila
 - Di Zhou
-- Yuning Wang
+- Pol Suárez Morales
 - Ting Zhu
 
 Special thanks for the guidance and support provided by:

build/include/precision.mod

@@ -1,6 +1,6 @@
 V34 :0x24 precision
 13 precision.f90 S624 0
-05/05/2025  14:50:44
+05/11/2025  16:18:05
 use iso_fortran_env private
 use iso_c_binding private
 use mpi private
@@ -23,9 +23,9 @@ D 79 23 6 1 11 58 0 0 0 0 0
  0 58 11 11 58 58
 D 82 26 686 8 685 7
 D 91 26 689 8 688 7
-S 624 24 0 0 0 9 1 0 3189 10005 0 A 0 0 0 0 B 0 0 0 0 0 0 0 0 0 0 0 0 21 0 0 0 0 0 0 0 0 8 0 0 0 0 0 0 precision
-S 626 23 0 0 0 6 1098 624 5016 4 0 A 0 0 0 0 B 400000 9 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 624 0 0 0 0 mpi_real
-S 627 23 0 0 0 6 1051 624 5025 4 0 A 0 0 0 0 B 400000 9 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 624 0 0 0 0 mpi_double_precision
+S 624 24 0 0 0 9 1 0 3189 10005 0 A 0 0 0 0 B 0 0 0 0 0 0 0 0 0 0 0 0 20 0 0 0 0 0 0 0 0 7 0 0 0 0 0 0 precision
+S 626 23 0 0 0 6 1098 624 5016 4 0 A 0 0 0 0 B 400000 8 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 624 0 0 0 0 mpi_real
+S 627 23 0 0 0 6 1051 624 5025 4 0 A 0 0 0 0 B 400000 8 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 624 0 0 0 0 mpi_double_precision
 S 630 3 0 0 0 6 1 1 0 0 0 A 0 0 0 0 B 0 0 0 0 0 0 0 0 0 8 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 6
 S 632 3 0 0 0 6 1 1 0 0 0 A 0 0 0 0 B 0 0 0 0 0 0 0 0 0 2 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 6
 S 637 3 0 0 0 6 1 1 0 0 0 A 0 0 0 0 B 0 0 0 0 0 0 0 0 0 4 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 6
@@ -43,14 +43,14 @@ R 723 6 45 iso_c_binding c_null_ptr$ac
 R 725 6 47 iso_c_binding c_null_funptr$ac
 R 726 26 48 iso_c_binding ==
 R 728 26 50 iso_c_binding !=
-S 753 3 0 0 0 6 1 1 0 0 0 A 0 0 0 0 B 0 0 0 0 0 0 0 0 0 13 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 6
+S 770 3 0 0 0 6 1 1 0 0 0 A 0 0 0 0 B 0 0 0 0 0 0 0 0 0 17 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 6
 R 1051 16 217 mpi mpi_double_precision
 R 1098 16 264 mpi mpi_real
-S 7875 16 1 0 0 6 7876 624 37013 4 400000 A 0 0 0 0 B 0 11 0 0 0 0 0 0 4 35 0 0 0 0 0 0 0 0 0 0 0 0 0 624 0 0 0 0 sp
-S 7876 16 0 0 0 6 7878 624 37016 4 400000 A 0 0 0 0 B 0 11 0 0 0 0 0 0 8 16 0 0 0 0 0 0 0 0 0 0 0 0 0 624 0 0 0 0 dp
-S 7878 16 0 0 0 6 7879 624 37019 800004 400000 A 0 0 0 0 B 0 11 0 0 0 0 0 0 8 16 0 0 0 0 0 0 0 0 0 0 0 0 0 624 0 0 0 0 i8
-S 7879 16 0 0 0 6 7880 624 37022 800004 400000 A 0 0 0 0 B 0 15 0 0 0 0 0 0 4 35 0 0 0 0 0 0 0 0 0 0 0 0 0 624 0 0 0 0 rp
-S 7880 16 0 0 0 6 0 624 37025 800004 400000 A 0 0 0 0 B 0 16 0 0 0 0 0 0 13 168 0 0 0 0 0 0 0 0 0 0 0 0 0 624 0 0 0 0 mpi_real_rp
+S 7687 16 0 0 0 6 7688 624 36396 4 400000 A 0 0 0 0 B 0 10 0 0 0 0 0 0 4 35 0 0 0 0 0 0 0 0 0 0 0 0 0 624 0 0 0 0 sp
+S 7688 16 1 0 0 6 7690 624 36399 4 400000 A 0 0 0 0 B 0 10 0 0 0 0 0 0 8 16 0 0 0 0 0 0 0 0 0 0 0 0 0 624 0 0 0 0 dp
+S 7690 16 0 0 0 6 7691 624 36402 800004 400000 A 0 0 0 0 B 0 10 0 0 0 0 0 0 8 16 0 0 0 0 0 0 0 0 0 0 0 0 0 624 0 0 0 0 i8
+S 7691 16 0 0 0 6 7692 624 36405 800004 400000 A 0 0 0 0 B 0 17 0 0 0 0 0 0 8 16 0 0 0 0 0 0 0 0 0 0 0 0 0 624 0 0 0 0 rp
+S 7692 16 0 0 0 6 0 624 36408 800004 400000 A 0 0 0 0 B 0 18 0 0 0 0 0 0 17 187 0 0 0 0 0 0 0 0 0 0 0 0 0 624 0 0 0 0 mpi_real_rp
 A 16 2 0 0 0 6 630 0 0 0 16 0 0 0 0 0 0 0 0 0 0 0
 A 33 2 0 0 0 6 632 0 0 0 33 0 0 0 0 0 0 0 0 0 0 0
 A 35 2 0 0 0 6 637 0 0 0 35 0 0 0 0 0 0 0 0 0 0 0
@@ -62,7 +62,7 @@ A 73 1 0 3 0 70 666 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0
 A 77 1 0 5 0 76 668 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0
 A 131 1 0 0 0 82 723 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0
 A 134 1 0 0 0 91 725 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0
-A 168 2 0 0 0 6 753 0 0 0 168 0 0 0 0 0 0 0 0 0 0 0
+A 187 2 0 0 0 6 770 0 0 0 187 0 0 0 0 0 0 0 0 0 0 0
 Z
 J 29 1 1
 V 61 58 7 0