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
|
/* -*- 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_ACCELERATOR_KOKKOS_H
#define LMP_ACCELERATOR_KOKKOS_H
// true interface to KOKKOS
// used when KOKKOS is installed
#ifdef LMP_KOKKOS
#include "kokkos.h"
#include "atom_kokkos.h"
#include "comm_kokkos.h"
#include "comm_tiled_kokkos.h"
#include "domain_kokkos.h"
#include "neighbor_kokkos.h"
#include "modify_kokkos.h"
#else
// dummy interface to KOKKOS
// needed for compiling when KOKKOS is not installed
#include "atom.h"
#include "comm_brick.h"
#include "comm_tiled.h"
#include "domain.h"
#include "neighbor.h"
#include "modify.h"
namespace LAMMPS_NS {
class KokkosLMP {
public:
int kokkos_exists;
int num_threads;
int ngpu;
int numa;
KokkosLMP(class LAMMPS *, int, char **) {kokkos_exists = 0;}
~KokkosLMP() {}
void accelerator(int, char **) {}
int neigh_list_kokkos(int) {return 0;}
int neigh_count(int) {return 0;}
};
class AtomKokkos : public Atom {
public:
tagint **k_special;
AtomKokkos(class LAMMPS *lmp) : Atom(lmp) {}
~AtomKokkos() {}
void sync(const ExecutionSpace space, unsigned int mask) {}
void modified(const ExecutionSpace space, unsigned int mask) {}
};
class CommKokkos : public CommBrick {
public:
CommKokkos(class LAMMPS *lmp) : CommBrick(lmp) {}
~CommKokkos() {}
};
class CommTiledKokkos : public CommTiled {
public:
CommTiledKokkos(class LAMMPS *lmp) : CommTiled(lmp) {}
CommTiledKokkos(class LAMMPS *lmp, Comm *oldcomm) : CommTiled(lmp,oldcomm) {}
~CommTiledKokkos() {}
};
class DomainKokkos : public Domain {
public:
DomainKokkos(class LAMMPS *lmp) : Domain(lmp) {}
~DomainKokkos() {}
};
class NeighborKokkos : public Neighbor {
public:
NeighborKokkos(class LAMMPS *lmp) : Neighbor(lmp) {}
~NeighborKokkos() {}
};
class ModifyKokkos : public Modify {
public:
ModifyKokkos(class LAMMPS *lmp) : Modify(lmp) {}
~ModifyKokkos() {}
};
class DAT {
public:
typedef double tdual_xfloat_1d;
typedef double tdual_FFT_SCALAR_1d;
typedef int t_int_1d;
typedef int tdual_int_2d;
};
}
#endif
#endif
|
