git-svn-id: svn://svn.icms.temple.edu/lammps-ro/trunk@8922 f3b2605a-c512-4ea7-a41b-209d697bcdaa

2012-10-08 15:29:55 +00:00
parent 856a237400
commit 14bcb1e6d0
184 changed files with 23096 additions and 0 deletions
--- a/lib/cuda/fix_set_force_cuda_kernel.cu
+++ b/lib/cuda/fix_set_force_cuda_kernel.cu
@ -0,0 +1,86 @@
+/* ----------------------------------------------------------------------
+   LAMMPS - Large-scale Atomic/Molecular Massively Parallel Simulator
+
+   Original Version:
+   http://lammps.sandia.gov, Sandia National Laboratories
+   Steve Plimpton, sjplimp@sandia.gov
+
+   See the README file in the top-level LAMMPS directory.
+
+   -----------------------------------------------------------------------
+
+   USER-CUDA Package and associated modifications:
+   https://sourceforge.net/projects/lammpscuda/
+
+   Christian Trott, christian.trott@tu-ilmenau.de
+   Lars Winterfeld, lars.winterfeld@tu-ilmenau.de
+   Theoretical Physics II, University of Technology Ilmenau, Germany
+
+   See the README file in the USER-CUDA directory.
+
+   This software is distributed under the GNU General Public License.
+------------------------------------------------------------------------- */
+
+extern __shared__ F_FLOAT sharedmem[];
+
+
+__global__ void Cuda_FixSetForceCuda_PostForce_Kernel(int groupbit, F_FLOAT xvalue, F_FLOAT yvalue, F_FLOAT zvalue, int flagx, int flagy, int flagz)
+{
+  int i = (blockIdx.x * gridDim.y + blockIdx.y) * blockDim.x + threadIdx.x;
+  sharedmem[threadIdx.x] = 0;
+  sharedmem[threadIdx.x + blockDim.x] = 0;
+  sharedmem[threadIdx.x + 2 * blockDim.x] = 0;
+
+  if(i < _nlocal)
+    if(_mask[i] & groupbit) {
+      sharedmem[threadIdx.x] = _f[i];
+      sharedmem[threadIdx.x + blockDim.x] = _f[i + 1 * _nmax];
+      sharedmem[threadIdx.x + 2 * blockDim.x] = _f[i + 2 * _nmax];
+
+      if(flagx) _f[i] = xvalue;
+
+      if(flagy) _f[i + 1 * _nmax] = yvalue;
+
+      if(flagz) _f[i + 2 * _nmax] = zvalue;
+    }
+
+
+  reduceBlock(sharedmem);
+  reduceBlock(&sharedmem[blockDim.x]);
+  reduceBlock(&sharedmem[2 * blockDim.x]);
+  F_FLOAT* buffer = (F_FLOAT*)_buffer;
+
+  if(threadIdx.x == 0) {
+    buffer[blockIdx.x * gridDim.y + blockIdx.y] = sharedmem[0];
+    buffer[blockIdx.x * gridDim.y + blockIdx.y + gridDim.x * gridDim.y] = sharedmem[blockDim.x];
+    buffer[blockIdx.x * gridDim.y + blockIdx.y + 2 * gridDim.x * gridDim.y] = sharedmem[2 * blockDim.x];
+  }
+}
+
+
+__global__ void Cuda_FixSetForceCuda_Reduce_FOriginal(int n, F_FLOAT* foriginal)
+{
+  int i = 0;
+  sharedmem[threadIdx.x] = 0;
+  F_FLOAT myforig = 0.0;
+  F_FLOAT* buf = (F_FLOAT*)_buffer;
+  buf = &buf[blockIdx.x * n];
+
+  while(i < n) {
+    sharedmem[threadIdx.x] = 0;
+
+    if(i + threadIdx.x < n)
+      sharedmem[threadIdx.x] = buf[i + threadIdx.x];
+
+    __syncthreads();
+    reduceBlock(sharedmem);
+    i += blockDim.x;
+
+    if(threadIdx.x == 0)
+      myforig += sharedmem[0];
+  }
+
+  if(threadIdx.x == 0)
+    foriginal[blockIdx.x] = myforig;
+}
+