Add conditional for nvfortran

minhqdao · Feb 26, 2024 · e5b4177 · e5b4177
1 parent 9f5b509
commit e5b4177
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/Makefile b/Makefile
@@ -20,7 +20,11 @@ all: $(STATIC)
 
 $(STATIC): $(SRC)
 		mkdir -p $(MOD-DIR) $(OBJ-DIR)
-		$(FC) $(FFLAGS) -c $(SRC) -J$(MOD-DIR) -o $(OBJ-DIR)/$(FILENAME).o
+		ifeq ($(FC),nvfortran)
+			$(FC) $(FFLAGS) -c $(SRC) -module $(MOD-DIR) -o $(OBJ-DIR)/$(FILENAME).o
+		else
+			$(FC) $(FFLAGS) -c $(SRC) -J$(MOD-DIR) -o $(OBJ-DIR)/$(FILENAME).o
+		endif
 		$(AR) $(ARFLAGS) $(STATIC) $(OBJ-DIR)/$(FILENAME).o
 
 test: $(TEST-SRC) $(STATIC)