From a1125d4c799c068e7a79f09eccf2811893a25076 Mon Sep 17 00:00:00 2001 From: CGH0S7 <776459475@qq.com> Date: Tue, 13 Jan 2026 23:52:44 +0800 Subject: [PATCH] try to build gpu version --- AMSS_NCKU_Input.py | 2 +- AMSS_NCKU_source/makefile.inc | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/AMSS_NCKU_Input.py b/AMSS_NCKU_Input.py index fe25a50..a4f1209 100755 --- a/AMSS_NCKU_Input.py +++ b/AMSS_NCKU_Input.py @@ -16,7 +16,7 @@ import numpy File_directory = "GW150914" ## output file directory Output_directory = "binary_output" ## binary data file directory ## The file directory name should not be too long -MPI_processes = 64 ## number of mpi processes used in the simulation +MPI_processes = 12 ## number of mpi processes used in the simulation GPU_Calculation = "no" ## Use GPU or not ## (prefer "no" in the current version, because the GPU part may have bugs when integrated in this Python interface) diff --git a/AMSS_NCKU_source/makefile.inc b/AMSS_NCKU_source/makefile.inc index af573ca..a512708 100755 --- a/AMSS_NCKU_source/makefile.inc +++ b/AMSS_NCKU_source/makefile.inc @@ -3,7 +3,7 @@ ##filein = -I/usr/include/ -I/usr/lib/x86_64-linux-gnu/openmpi/include/ -I/usr/lib/x86_64-linux-gnu/openmpi/lib/ -I/usr/lib/gcc/x86_64-linux-gnu/11/ -I/usr/include/c++/11/ -I/usr/lib/cuda/include -LDLIBS = -lmpi -lgfortran +LDLIBS = -lmpi -lgfortran -lcudart ##LDLIBS = -L/usr/lib/x86_64-linux-gnu -L/usr/lib64 -L/usr/lib/gcc/x86_64-linux-gnu/11 -lgfortran -L/usr/lib/cuda/lib64 -lcudart -lmpi -lgfortran CXXAPPFLAGS = -O3 -Wno-deprecated -Dfortran3 -Dnewc @@ -16,6 +16,6 @@ CC = gcc CLINKER = mpic++ Cu = nvcc -CUDA_LIB_PATH = -L/usr/lib/cuda/lib64 -I/usr/include -I/usr/lib/cuda/include +CUDA_LIB_PATH = -L/opt/nvidia/hpc_sdk/Linux_x86_64/24.5/cuda/lib64 -I/usr/include -I/opt/nvidia/hpc_sdk/Linux_x86_64/24.5/cuda/include #CUDA_APP_FLAGS = -c -g -O3 --ptxas-options=-v -arch compute_13 -code compute_13,sm_13 -Dfortran3 -Dnewc CUDA_APP_FLAGS = -c -g -O3 --ptxas-options=-v -Dfortran3 -Dnewc