1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
/* -*- c++ -*- ----------------------------------------------------------
LAMMPS - Large-scale Atomic/Molecular Massively Parallel Simulator
http://lammps.sandia.gov, Sandia National Laboratories
Steve Plimpton, sjplimp@sandia.gov
Copyright (2003) Sandia Corporation. Under the terms of Contract
DE-AC04-94AL85000 with Sandia Corporation, the U.S. Government retains
certain rights in this software. This software is distributed under
the GNU General Public License.
See the README file in the top-level LAMMPS directory.
------------------------------------------------------------------------- */
#ifndef LMP_COMM_KOKKOS_H
#define LMP_COMM_KOKKOS_H
#include "comm_brick.h"
#include "kokkos_type.h"
namespace LAMMPS_NS {
class CommKokkos : public CommBrick {
public:
bool exchange_comm_classic;
bool forward_comm_classic;
bool reverse_comm_classic;
bool exchange_comm_on_host;
bool forward_comm_on_host;
bool reverse_comm_on_host;
CommKokkos(class LAMMPS *);
~CommKokkos();
void init();
void forward_comm(int dummy = 0); // forward comm of atom coords
void reverse_comm(); // reverse comm of atom coords
void exchange(); // move atoms to new procs
void borders(); // setup list of atoms to comm
void forward_comm_pair(class Pair *); // forward comm from a Pair
void reverse_comm_pair(class Pair *); // reverse comm from a Pair
void forward_comm_fix(class Fix *, int size=0); // forward comm from a Fix
void reverse_comm_fix(class Fix *, int size=0); // reverse comm from a Fix
void forward_comm_compute(class Compute *); // forward from a Compute
void reverse_comm_compute(class Compute *); // reverse from a Compute
void forward_comm_dump(class Dump *); // forward comm from a Dump
void reverse_comm_dump(class Dump *); // reverse comm from a Dump
template<class DeviceType> void forward_comm_device(int dummy);
template<class DeviceType> void reverse_comm_device();
template<class DeviceType> void forward_comm_pair_device(Pair *pair);
template<class DeviceType> void exchange_device();
template<class DeviceType> void borders_device();
protected:
DAT::tdual_int_2d k_sendlist;
DAT::tdual_int_scalar k_total_send;
DAT::tdual_xfloat_2d k_buf_send,k_buf_recv;
DAT::tdual_int_2d k_exchange_lists;
DAT::tdual_int_1d k_exchange_sendlist,k_exchange_copylist,k_sendflag;
DAT::tdual_int_scalar k_count;
//double *buf_send; // send buffer for all comm
//double *buf_recv; // recv buffer for all comm
DAT::tdual_int_2d k_swap;
DAT::tdual_int_2d k_swap2;
DAT::tdual_int_2d k_pbc;
DAT::tdual_int_1d k_pbc_flag;
DAT::tdual_int_1d k_g2l;
DAT::tdual_int_1d k_firstrecv;
DAT::tdual_int_1d k_sendnum_scan;
int totalsend;
int max_buf_pair;
DAT::tdual_xfloat_1d k_buf_send_pair;
DAT::tdual_xfloat_1d k_buf_recv_pair;
void grow_buf_pair(int);
void grow_send(int, int);
void grow_recv(int);
void grow_send_kokkos(int, int, ExecutionSpace space = Host);
void grow_recv_kokkos(int, ExecutionSpace space = Host);
void grow_list(int, int);
void grow_swap(int);
void copy_swap_info();
};
}
#endif
/* ERROR/WARNING messages:
E: Ghost velocity forward comm not yet implemented with Kokkos
This is a current restriction.
W: Fixes cannot yet send data in Kokkos communication, switching to classic communication
This is a current restriction with Kokkos.
W: Required border comm not yet implemented in Kokkos communication, switching to classic communication
There are various limitations in the communication options supported
by Kokkos.
E: Required border comm not yet implemented with Kokkos
There are various limitations in the communication options supported
by Kokkos.
*/