#ifdef ATOM_CLASS
AtomStyle(bond/kk,AtomVecBondKokkos)
AtomStyle(bond/kk/device,AtomVecBondKokkos)
AtomStyle(bond/kk/host,AtomVecBondKokkos)
#else
#ifndef LMP_ATOM_VEC_BOND_KOKKOS_H
#define LMP_ATOM_VEC_BOND_KOKKOS_H
#include "atom_vec_kokkos.h"
namespace LAMMPS_NS {
class AtomVecBondKokkos : public AtomVecKokkos {
public:
AtomVecBondKokkos(class LAMMPS *);
virtual ~AtomVecBondKokkos() {}
void grow(int);
void copy(int, int, int);
int pack_border(int, int *, double *, int, int *);
int pack_border_vel(int, int *, double *, int, int *);
int pack_border_hybrid(int, int *, double *);
void unpack_border(int, int, double *);
void unpack_border_vel(int, int, double *);
int unpack_border_hybrid(int, int, double *);
int pack_exchange(int, double *);
int unpack_exchange(double *);
int size_restart();
int pack_restart(int, double *);
int unpack_restart(double *);
void create_atom(int, double *);
void data_atom(double *, tagint, char **);
int data_atom_hybrid(int, char **);
void pack_data(double **);
int pack_data_hybrid(int, double *);
void write_data(FILE *, int, double **);
int write_data_hybrid(FILE *, double *);
bigint memory_usage();
void grow_reset();
int pack_border_kokkos(int n, DAT::tdual_int_2d k_sendlist,
DAT::tdual_xfloat_2d buf,int iswap,
int pbc_flag, int *pbc, ExecutionSpace space);
void unpack_border_kokkos(const int &n, const int &nfirst,
const DAT::tdual_xfloat_2d &buf,
ExecutionSpace space);
int pack_exchange_kokkos(const int &nsend,DAT::tdual_xfloat_2d &buf,
DAT::tdual_int_1d k_sendlist,
DAT::tdual_int_1d k_copylist,
ExecutionSpace space, int dim,
X_FLOAT lo, X_FLOAT hi);
int unpack_exchange_kokkos(DAT::tdual_xfloat_2d &k_buf, int nrecv,
int nlocal, int dim, X_FLOAT lo, X_FLOAT hi,
ExecutionSpace space);
void sync(ExecutionSpace space, unsigned int mask);
void modified(ExecutionSpace space, unsigned int mask);
void sync_overlapping_device(ExecutionSpace space, unsigned int mask);
protected:
tagint *tag;
int *type,*mask;
imageint *image;
double **x,**v,**f;
tagint *molecule;
int **nspecial;
tagint **special;
int *num_bond;
int **bond_type;
tagint **bond_atom;
DAT::t_tagint_1d d_tag;
DAT::t_int_1d d_type, d_mask;
HAT::t_tagint_1d h_tag;
HAT::t_int_1d h_type, h_mask;
DAT::t_imageint_1d d_image;
HAT::t_imageint_1d h_image;
DAT::t_x_array d_x;
DAT::t_v_array d_v;
DAT::t_f_array d_f;
DAT::t_tagint_1d d_molecule;
DAT::t_int_2d d_nspecial;
DAT::t_tagint_2d d_special;
DAT::t_int_1d d_num_bond;
DAT::t_int_2d d_bond_type;
DAT::t_tagint_2d d_bond_atom;
HAT::t_tagint_1d h_molecule;
HAT::t_int_2d h_nspecial;
HAT::t_tagint_2d h_special;
HAT::t_int_1d h_num_bond;
HAT::t_int_2d h_bond_type;
HAT::t_tagint_2d h_bond_atom;
DAT::tdual_int_1d k_count;
};
}
#endif
#endif