OilfieldToolsNet/LBPM
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update cpu ion valence

Browse Source
This commit is contained in:
James McClure
2022-04-26 06:37:08 -04:00
parent 429413ce3b
commit 3e82370d6c
2 changed files with 17 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
common/Membrane.cpp
View File

@@ -5,22 +5,17 @@
Membrane::Membrane(std::shared_ptr <Domain> Dm, int *dvcNeighborList, int Nsites) { Membrane::Membrane(std::shared_ptr <Domain> Dm, int *dvcNeighborList, int Nsites) {
Np = Nsites;
initialNeighborList = new int[18*Np];
ScaLBL_AllocateDeviceMemory((void **)&NeighborList, 18*Np*sizeof(int));
Lock=false; // unlock the communicator
//...................................................................................... //......................................................................................
// Create a separate copy of the communicator for the device // Create a separate copy of the communicator for the device
MPI_COMM_SCALBL = Dm->Comm.dup(); MPI_COMM_SCALBL = Dm->Comm.dup();
Np = Nsites;
if (Dm->rank() == 0){
printf("**** Creating membrane data structure ****** \n");
printf(" Number of active lattice sites (rank = %i): %i \n",Dm->rank(), Np);
}
initialNeighborList = new int[18*Np];
ScaLBL_AllocateDeviceMemory((void **)&NeighborList, 18*Np*sizeof(int));
Lock=false; // unlock the communicator
ScaLBL_CopyToHost(initialNeighborList, dvcNeighborList, 18*Np*sizeof(int)); ScaLBL_CopyToHost(initialNeighborList, dvcNeighborList, 18*Np*sizeof(int));
Dm->Comm.barrier(); Dm->Comm.barrier();
ScaLBL_CopyToDevice(NeighborList, initialNeighborList, 18*Np*sizeof(int)); ScaLBL_CopyToDevice(NeighborList, initialNeighborList, 18*Np*sizeof(int));
/* Copy communication lists */ /* Copy communication lists */
//...................................................................................... //......................................................................................
@@ -91,6 +86,11 @@ Membrane::Membrane(std::shared_ptr <Domain> Dm, int *dvcNeighborList, int Nsites
recvCount_YZ=Dm->recvCount("YZ"); recvCount_YZ=Dm->recvCount("YZ");
recvCount_XZ=Dm->recvCount("XZ"); recvCount_XZ=Dm->recvCount("XZ");
if (rank == 0){
printf("**** Creating membrane data structure ****** \n");
printf(" Number of active lattice sites (rank = %i): %i \n",rank, Np);
}
iproc = Dm->iproc(); iproc = Dm->iproc();
jproc = Dm->jproc(); jproc = Dm->jproc();
kproc = Dm->kproc(); kproc = Dm->kproc();
@@ -99,7 +99,7 @@ Membrane::Membrane(std::shared_ptr <Domain> Dm, int *dvcNeighborList, int Nsites
nprocz = Dm->nprocz(); nprocz = Dm->nprocz();
//BoundaryCondition = Dm->BoundaryCondition; //BoundaryCondition = Dm->BoundaryCondition;
//...................................................................................... //......................................................................................
if (rank==0) printf(" Create Membrane: copying communication structures...\n");
ScaLBL_AllocateZeroCopy((void **) &sendbuf_x, 2*5*sendCount_x*sizeof(double)); // Allocate device memory ScaLBL_AllocateZeroCopy((void **) &sendbuf_x, 2*5*sendCount_x*sizeof(double)); // Allocate device memory
ScaLBL_AllocateZeroCopy((void **) &sendbuf_X, 2*5*sendCount_X*sizeof(double)); // Allocate device memory ScaLBL_AllocateZeroCopy((void **) &sendbuf_X, 2*5*sendCount_X*sizeof(double)); // Allocate device memory
ScaLBL_AllocateZeroCopy((void **) &sendbuf_y, 2*5*sendCount_y*sizeof(double)); // Allocate device memory ScaLBL_AllocateZeroCopy((void **) &sendbuf_y, 2*5*sendCount_y*sizeof(double)); // Allocate device memory
@@ -1323,7 +1323,7 @@ void Membrane::RecvD3Q7AA(double *dist){
// Unpack the distributions on the device // Unpack the distributions on the device
//................................................................................... //...................................................................................
//...Unpacking for x face(q=2)................................ //...Unpacking for x face(q=2)................................
/* ScaLBL_D3Q7_Membrane_Unpack(2,dvcRecvDist_x, dvcRecvLinks_x,0,linkCount_x[0],recvCount_x,recvbuf_x,dist,Np,coefficient_x); ScaLBL_D3Q7_Membrane_Unpack(2,dvcRecvDist_x, dvcRecvLinks_x,0,linkCount_x[0],recvCount_x,recvbuf_x,dist,Np,coefficient_x);
//................................................................................... //...................................................................................
//...Packing for X face(q=1)................................ //...Packing for X face(q=1)................................
ScaLBL_D3Q7_Membrane_Unpack(1,dvcRecvDist_X, dvcRecvLinks_X,0,linkCount_X[0],recvCount_X,recvbuf_X,dist,Np,coefficient_X); ScaLBL_D3Q7_Membrane_Unpack(1,dvcRecvDist_X, dvcRecvLinks_X,0,linkCount_X[0],recvCount_X,recvbuf_X,dist,Np,coefficient_X);
@@ -1339,7 +1339,7 @@ void Membrane::RecvD3Q7AA(double *dist){
//...Packing for Z face(q=5)................................ //...Packing for Z face(q=5)................................
ScaLBL_D3Q7_Membrane_Unpack(5,dvcRecvDist_Z, dvcRecvLinks_Z,0,linkCount_Z[0],recvCount_Z,recvbuf_Z,dist,Np,coefficient_Z); ScaLBL_D3Q7_Membrane_Unpack(5,dvcRecvDist_Z, dvcRecvLinks_Z,0,linkCount_Z[0],recvCount_Z,recvbuf_Z,dist,Np,coefficient_Z);
//.................................................................................. //..................................................................................
*/
//................................................................................... //...................................................................................
Lock=false; // unlock the communicator after communications complete Lock=false; // unlock the communicator after communications complete
//................................................................................... //...................................................................................

2
cpu/Ion.cpp
View File

@@ -482,7 +482,7 @@ extern "C" void ScaLBL_D3Q7_Ion_Init_FromFile(double *dist, double *Den,
extern "C" void ScaLBL_D3Q7_Ion_ChargeDensity(double *Den, extern "C" void ScaLBL_D3Q7_Ion_ChargeDensity(double *Den,
double *ChargeDensity, double *ChargeDensity,
int IonValence, int ion_component, double IonValence, int ion_component,
int start, int finish, int Np) { int start, int finish, int Np) {
int n; int n;