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
/* -*- 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.
------------------------------------------------------------------------- */
#ifdef FIX_CLASS
FixStyle(OMP,FixOMP)
#else
#ifndef LMP_FIX_OMP_H
#define LMP_FIX_OMP_H
#include "fix.h"
namespace LAMMPS_NS {
class ThrData;
class FixOMP : public Fix {
friend class ThrOMP;
friend class RespaOMP;
public:
FixOMP(class LAMMPS *, int, char **);
virtual ~FixOMP();
virtual int setmask();
virtual void init();
virtual void setup(int);
virtual void min_setup(int flag) { setup(flag); }
virtual void pre_force(int);
virtual void setup_pre_force(int vflag) { pre_force(vflag); }
virtual void min_setup_pre_force(int vflag) { pre_force(vflag); }
virtual void min_pre_force(int vflag) { pre_force(vflag); }
virtual void setup_pre_force_respa(int vflag,int) { pre_force(vflag); }
virtual void pre_force_respa(int vflag,int,int) { pre_force(vflag); }
virtual double memory_usage();
protected:
ThrData **thr;
void *last_omp_style; // pointer to the style that needs
// to do the general force reduction
void *last_pair_hybrid; // pointer to the pair style that needs
// to call virial_fdot_compute()
// signal that an /omp style did the force reduction. needed by respa/omp
void did_reduce() { _reduced = true; }
public:
ThrData *get_thr(int tid) { return thr[tid]; }
int get_nthr() const { return _nthr; }
bool get_neighbor() const { return _neighbor; }
bool get_mixed() const { return _mixed; }
bool get_reduced() const { return _reduced; }
private:
int _nthr; // number of currently active ThrData objects
bool _neighbor; // en/disable threads for neighbor list construction
bool _mixed; // whether to prefer mixed precision compute kernels
bool _reduced; // whether forces have been reduced for this step
void set_neighbor_omp();
};
}
#endif
#endif