1
/* -*- c++ -*- ----------------------------------------------------------
2
LAMMPS - Large-scale Atomic/Molecular Massively Parallel Simulator
3
http://lammps.sandia.gov, Sandia National Laboratories
4
Steve Plimpton, sjplimp@sandia.gov
6
Copyright (2003) Sandia Corporation. Under the terms of Contract
7
DE-AC04-94AL85000 with Sandia Corporation, the U.S. Government retains
8
certain rights in this software. This software is distributed under
9
the GNU General Public License.
11
See the README file in the top-level LAMMPS directory.
12
------------------------------------------------------------------------- */
16
AtomStyle(charge/kk,AtomVecChargeKokkos)
20
#ifndef LMP_ATOM_VEC_CHARGE_KOKKOS_H
21
#define LMP_ATOM_VEC_CHARGE_KOKKOS_H
23
#include "atom_vec_kokkos.h"
24
#include "kokkos_type.h"
28
class AtomVecChargeKokkos : public AtomVecKokkos {
30
AtomVecChargeKokkos(class LAMMPS *);
31
virtual ~AtomVecChargeKokkos() {}
33
void copy(int, int, int);
34
int pack_comm(int, int *, double *, int, int *);
35
int pack_comm_vel(int, int *, double *, int, int *);
36
void unpack_comm(int, int, double *);
37
void unpack_comm_vel(int, int, double *);
38
int pack_reverse(int, int, double *);
39
void unpack_reverse(int, int *, double *);
40
int pack_border(int, int *, double *, int, int *);
41
int pack_border_vel(int, int *, double *, int, int *);
42
int pack_border_hybrid(int, int *, double *);
43
void unpack_border(int, int, double *);
44
void unpack_border_vel(int, int, double *);
45
int unpack_border_hybrid(int, int, double *);
46
int pack_exchange(int, double *);
47
int unpack_exchange(double *);
49
int pack_restart(int, double *);
50
int unpack_restart(double *);
51
void create_atom(int, double *);
52
void data_atom(double *, tagint, char **);
53
int data_atom_hybrid(int , char **);
54
void pack_data(double **);
55
int pack_data_hybrid(int, double *);
56
void write_data(FILE *, int, double **);
57
int write_data_hybrid(FILE *, double *);
58
bigint memory_usage();
61
int pack_comm_kokkos(const int &n, const DAT::tdual_int_2d &k_sendlist,
63
const DAT::tdual_xfloat_2d &buf,
64
const int &pbc_flag, const int pbc[]);
65
void unpack_comm_kokkos(const int &n, const int &nfirst,
66
const DAT::tdual_xfloat_2d &buf);
67
int pack_comm_self(const int &n, const DAT::tdual_int_2d &list,
68
const int & iswap, const int nfirst,
69
const int &pbc_flag, const int pbc[]);
70
int pack_border_kokkos(int n, DAT::tdual_int_2d k_sendlist,
71
DAT::tdual_xfloat_2d buf,int iswap,
72
int pbc_flag, int *pbc, ExecutionSpace space);
73
void unpack_border_kokkos(const int &n, const int &nfirst,
74
const DAT::tdual_xfloat_2d &buf,
75
ExecutionSpace space);
76
int pack_exchange_kokkos(const int &nsend,DAT::tdual_xfloat_2d &buf,
77
DAT::tdual_int_1d k_sendlist,
78
DAT::tdual_int_1d k_copylist,
79
ExecutionSpace space, int dim,
80
X_FLOAT lo, X_FLOAT hi);
81
int unpack_exchange_kokkos(DAT::tdual_xfloat_2d &k_buf, int nrecv,
82
int nlocal, int dim, X_FLOAT lo, X_FLOAT hi,
83
ExecutionSpace space);
85
void sync(ExecutionSpace space, unsigned int mask);
86
void modified(ExecutionSpace space, unsigned int mask);
96
DAT::t_tagint_1d d_tag;
97
HAT::t_tagint_1d h_tag;
99
DAT::t_int_1d d_type, d_mask;
100
HAT::t_int_1d h_type, h_mask;
102
DAT::t_imageint_1d d_image;
103
HAT::t_imageint_1d h_image;
116
DAT::tdual_int_1d k_count;
124
/* ERROR/WARNING messages: