libhoomd/cuda/TwoStepNVTRigidGPU.cuh

   1 /*
   2 Highly Optimized Object-oriented Many-particle Dynamics -- Blue Edition
   3 (HOOMD-blue) Open Source Software License Copyright 2009-2014 The Regents of
   4 the University of Michigan All rights reserved.
   5
   6 HOOMD-blue may contain modifications ("Contributions") provided, and to which
   7 copyright is held, by various Contributors who have granted The Regents of the
   8 University of Michigan the right to modify and/or distribute such Contributions.
   9
  10 You may redistribute, use, and create derivate works of HOOMD-blue, in source
  11 and binary forms, provided you abide by the following conditions:
  12
  13 * Redistributions of source code must retain the above copyright notice, this
  14 list of conditions, and the following disclaimer both in the code and
  15 prominently in any materials provided with the distribution.
  16
  17 * Redistributions in binary form must reproduce the above copyright notice, this
  18 list of conditions, and the following disclaimer in the documentation and/or
  19 other materials provided with the distribution.
  20
  21 * All publications and presentations based on HOOMD-blue, including any reports
  22 or published results obtained, in whole or in part, with HOOMD-blue, will
  23 acknowledge its use according to the terms posted at the time of submission on:
  24 http://codeblue.umich.edu/hoomd-blue/citations.html
  25
  26 * Any electronic documents citing HOOMD-Blue will link to the HOOMD-Blue website:
  27 http://codeblue.umich.edu/hoomd-blue/
  28
  29 * Apart from the above required attributions, neither the name of the copyright
  30 holder nor the names of HOOMD-blue's contributors may be used to endorse or
  31 promote products derived from this software without specific prior written
  32 permission.
  33
  34 Disclaimer
  35
  36 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDER AND CONTRIBUTORS ``AS IS'' AND
  37 ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  38 WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, AND/OR ANY
  39 WARRANTIES THAT THIS SOFTWARE IS FREE OF INFRINGEMENT ARE DISCLAIMED.
  40
  41 IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
  42 INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
  43 BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  44 DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  45 LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
  46 OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
  47 ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  48 */
  49
  50
  51 // Maintainer: ndtrung
  52
  53 /*! \file TwoStepNVTRigidGPU.cuh
  54     \brief Declares GPU kernel code for NVT rigid body integration on the GPU. Used by TwoStepNVTRigidGPU.
  55 */
  56
  57 #include "ParticleData.cuh"
  58 #include "RigidData.cuh"
  59 #include "TwoStepNVERigidGPU.cuh"
  60
  61 #ifndef __TWO_STEP_NVT_RIGID_CUH__
  62 #define __TWO_STEP_NVT_RIGID_CUH__
  63
  64 /*! Thermostat data structure
  65 */
  66 struct gpu_nvt_rigid_data
  67     {
  68     unsigned int n_bodies;  //!< Number of rigid bodies
  69
  70     Scalar  eta_dot_t0;      //!< Thermostat translational velocity
  71     Scalar  eta_dot_r0;      //!< Thermostat rotational velocity
  72
  73     Scalar *partial_Ksum_t;  //!< NBlocks elements, each is a partial sum of m*v^2
  74     Scalar *partial_Ksum_r;  //!< NBlocks elements, each is a partial sum of L*w^2
  75     Scalar *Ksum_t;          //!< fully reduced Ksum_t on one GPU
  76     Scalar *Ksum_r;          //!< fully reduced Ksum_r on one GPU
  77     };
  78
  79 //! Kernel driver for the first part of the NVT update called by TwoStepNVTRigidGPU
  80 cudaError_t gpu_nvt_rigid_step_one( const gpu_rigid_data_arrays& rigid_data,
  81                                         unsigned int *d_group_members,
  82                                         unsigned int group_size,
  83                                         Scalar4 *d_net_force,
  84                                         const BoxDim& box,
  85                                         const gpu_nvt_rigid_data &nvt_rdata,
  86                                         Scalar deltaT);
  87
  88 //! Kernel driver for the Ksum reduction final pass called by TwoStepNVTRigidGPU
  89 cudaError_t gpu_nvt_rigid_reduce_ksum(const gpu_nvt_rigid_data &nvt_rdata);
  90
  91 //! Kernel driver for the second part of the NVT update called by TwoStepNVTRigidGPU
  92 cudaError_t gpu_nvt_rigid_step_two( const gpu_rigid_data_arrays& rigid_data,
  93                                     unsigned int *d_group_members,
  94                                     unsigned int group_size,
  95                                     Scalar4 *d_net_force,
  96                                     Scalar *d_net_virial,
  97                                     const BoxDim& box,
  98                                     const gpu_nvt_rigid_data &nvt_rdata,
  99                                     Scalar deltaT);
 100
 101 #endif // __TWO_STEP_NVT_RIGID_CUH__