1 /* -*- mode: c; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4; c-file-style: "stroustrup"; -*-
4 * This source code is part of
8 * GROningen MAchine for Chemical Simulations
10 * Written by David van der Spoel, Erik Lindahl, Berk Hess, and others.
11 * Copyright (c) 1991-2000, University of Groningen, The Netherlands.
12 * Copyright (c) 2001-2012, The GROMACS development team,
13 * check out http://www.gromacs.org for more information.
15 * This program is free software; you can redistribute it and/or
16 * modify it under the terms of the GNU General Public License
17 * as published by the Free Software Foundation; either version 2
18 * of the License, or (at your option) any later version.
20 * If you want to redistribute modifications, please consider that
21 * scientific software is very special. Version control is crucial -
22 * bugs must be traceable. We will be happy to consider code for
23 * inclusion in the official distribution, but derived work must not
24 * be called official GROMACS. Details are found in the README & COPYING
25 * files - if they are missing, get the official version at www.gromacs.org.
27 * To help us fund GROMACS development, we humbly ask that you cite
28 * the papers on the package - you can find them in the top README file.
30 * For more info, check our website at http://www.gromacs.org
33 * Gallium Rubidium Oxygen Manganese Argon Carbon Silicon
36 #ifndef VECTYPE_OPS_CUH
37 #define VECTYPE_OPS_CUH
40 inline __host__ __device__ float3 make_float3(float s)
42 return make_float3(s, s, s);
44 inline __host__ __device__ float3 make_float3(float4 a)
46 return make_float3(a.x, a.y, a.z);
48 inline __host__ __device__ float3 operator-(float3 &a)
50 return make_float3(-a.x, -a.y, -a.z);
52 inline __host__ __device__ float3 operator+(float3 a, float3 b)
54 return make_float3(a.x + b.x, a.y + b.y, a.z + b.z);
56 inline __host__ __device__ float3 operator-(float3 a, float3 b)
58 return make_float3(a.x - b.x, a.y - b.y, a.z - b.z);
60 inline __host__ __device__ float3 operator*(float3 a, float k)
62 return make_float3(k * a.x, k * a.y, k * a.z);
64 inline __host__ __device__ float3 operator*(float k, float3 a)
66 return make_float3(k * a.x, k * a.y, k * a.z);
68 inline __host__ __device__ void operator+=(float3 &a, float3 b)
70 a.x += b.x; a.y += b.y; a.z += b.z;
72 inline __host__ __device__ void operator+=(float3 &a, float4 b)
74 a.x += b.x; a.y += b.y; a.z += b.z;
76 inline __host__ __device__ void operator-=(float3 &a, float3 b)
78 a.x -= b.x; a.y -= b.y; a.z -= b.z;
80 inline __host__ __device__ float norm(float3 a)
82 return sqrt(a.x * a.x + a.y * a.y + a.z * a.z);
84 inline __host__ __device__ float norm2(float3 a)
86 return (a.x * a.x + a.y * a.y + a.z * a.z);
88 inline __host__ __device__ float dist3(float3 a, float3 b)
92 inline __host__ __device__ float3 operator*(float3 a, float3 b)
94 return make_float3(a.x * b.x, a.y * b.y, a.z * b.z);
96 inline __host__ __device__ void operator*=(float3 &a, float3 b)
98 a.x *= b.x; a.y *= b.y; a.z *= b.z;
100 inline __device__ void atomicAdd(float3 *addr, float3 val)
102 atomicAdd(&addr->x, val.x);
103 atomicAdd(&addr->y, val.y);
104 atomicAdd(&addr->z, val.z);
106 /****************************************************************/
109 inline __host__ __device__ float4 make_float4(float s)
111 return make_float4(s, s, s, s);
113 inline __host__ __device__ float4 make_float4(float3 a)
115 return make_float4(a.x, a.y, a.z, 0.0f);
117 inline __host__ __device__ float4 operator+(float4 a, float4 b)
119 return make_float4(a.x + b.x, a.y + b.y, a.z + b.z, a.w + b.w);
121 inline __host__ __device__ float4 operator+(float4 a, float3 b)
123 return make_float4(a.x + b.x, a.y + b.y, a.z + b.z, a.w);
125 inline __host__ __device__ float4 operator-(float4 a, float4 b)
127 return make_float4(a.x - b.x, a.y - b.y, a.z - b.z, a.w - b.w);
129 inline __host__ __device__ float4 operator*(float4 a, float k)
131 return make_float4(k * a.x, k * a.y, k * a.z, k * a.w);
133 inline __host__ __device__ void operator+=(float4 &a, float4 b)
135 a.x += b.x; a.y += b.y; a.z += b.z; a.w += b.w;
137 inline __host__ __device__ void operator+=(float4 &a, float3 b)
139 a.x += b.x; a.y += b.y; a.z += b.z;
141 inline __host__ __device__ void operator-=(float4 &a, float3 b)
143 a.x -= b.x; a.y -= b.y; a.z -= b.z;
146 inline __host__ __device__ float norm(float4 a)
148 return sqrt(a.x * a.x + a.y * a.y + a.z * a.z + a.w * a.w);
151 inline __host__ __device__ float dist3(float4 a, float4 b)
156 #endif /* VECTYPE_OPS_CUH */