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
/* -*- 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_NPAIR_OMP_H
#define LMP_NPAIR_OMP_H
#if defined(_OPENMP)
#include <omp.h>
#endif
#include "comm.h"
#include "modify.h"
#include "timer.h"
#include "fix_omp.h"
#include "thr_data.h"
namespace LAMMPS_NS {
// these macros hide some ugly and redundant OpenMP related stuff
#if defined(_OPENMP)
// get access to number of threads and per-thread data structures via FixOMP
#define NPAIR_OMP_INIT \
const int nthreads = comm->nthreads; \
const int ifix = modify->find_fix("package_omp")
// get thread id and then assign each thread a fixed chunk of atoms
#define NPAIR_OMP_SETUP(num) \
{ \
const int tid = omp_get_thread_num(); \
const int idelta = 1 + num/nthreads; \
const int ifrom = tid*idelta; \
const int ito = ((ifrom + idelta) > num) \
? num : (ifrom+idelta); \
FixOMP *fix = static_cast<FixOMP *>(modify->fix[ifix]); \
ThrData *thr = fix->get_thr(tid); \
thr->timer(Timer::START);
#define NPAIR_OMP_CLOSE \
thr->timer(Timer::NEIGH); \
}
#else /* !defined(_OPENMP) */
#define NPAIR_OMP_INIT
#define NPAIR_OMP_SETUP(num) \
const int tid = 0; \
const int ifrom = 0; \
const int ito = num
#define NPAIR_OMP_CLOSE
#endif
}
#endif