geometry_omp.h

Go to the documentation of this file.

 
#ifndef GEOMETRY_OMP_H
#define GEOMETRY_OMP_H
 
#include "hr_omp.h"
 
#define _PIECE_FOR(type, ip) for (int ip = 0; ip < (type)->local_master_pieces; ip++)
 
#define _SITE_FOR_RED(type, ip, is, redop1, redop2) \
    _OMP_PRAGMA(_omp_parallel)                      \
    _OMP_PRAGMA(_omp_for redop1 redop2)             \
    for (size_t is = (type)->master_start[ip]; is <= (type)->master_end[ip]; is++)
 
#define _SITE_FOR(type, ip, is) _SITE_FOR_RED(type, ip, is, nowait, )
 
#define _SITE_FOR_SUM(type, ip, is, ...) _SITE_FOR_RED(type, ip, is, _omp_sum(__VA_ARGS__), )
 
#define _SITE_FOR_MAX(type, ip, is, ...) _SITE_FOR_RED(type, ip, is, _omp_max(__VA_ARGS__), )
 
#define _SITE_FOR_MIN(type, ip, is, ...) _SITE_FOR_RED(type, ip, is, _omp_min(__VA_ARGS__), )
 
#define _MASTER_FOR_RED(type, is, redop1, redop2) \
    _PIECE_FOR((type), _master_for_ip_##is)       \
        _SITE_FOR_RED((type), _master_for_ip_##is, is, redop1, redop2)
 
#define _MASTER_FOR(type, is) _MASTER_FOR_RED(type, is, , )
 
#define _MASTER_FOR_SUM(type, is, ...) _MASTER_FOR_RED(type, is, _omp_sum(__VA_ARGS__), )
 
#define _MASTER_FOR_MAX(type, is, ...) _MASTER_FOR_RED(type, is, _omp_max(__VA_ARGS__), )
 
#define _MASTER_FOR_MIN(type, is, ...) _MASTER_FOR_RED(type, is, _omp_min(__VA_ARGS__), )
 
#define _FUSE_FOR_RED(type, ip, is, redop1, redop2) \
    _OMP_PRAGMA(_omp_parallel)                      \
    _OMP_PRAGMA(_omp_for redop1 redop2)             \
    for (int ip = 0; ip < type->fuse_gauge_size; ip++)
 
#define _FUSE_MASTER_FOR_RED(type, is, redop1, redop2) _FUSE_FOR_RED((type), _fuse_master_for_ip_##is, is, redop1, redop2)
 
#define _ONE_SPINOR_FOR_RED(s, redop1, redop2) _MASTER_FOR_RED((s)->type, _spinor_for_is, redop1, redop2)
 
#define _ONE_SITE_FOR_RED(s, redop1, redop2) _MASTER_FOR_RED((s)->type, _site_for_is, redop1, redop2)
 
#define _ONE_SPINOR_FOR_SUM(s, ...) _ONE_SPINOR_FOR_RED(s, _omp_sum(__VA_ARGS__), )
 
#define _ONE_SITE_FOR_SUM(s, ...) _ONE_SITE_FOR_RED(s, _omp_sum(__VA_ARGS__), )
 
#define _ONE_SITE_FOR_MAX(s, ...) _ONE_SITE_FOR_RED(s, _omp_max(__VA_ARGS__), )
 
#define _TWO_SPINORS_FOR_RED(s1, s2, redop1, redop2) \
    _TWO_SPINORS_MATCHING(s1, s2);                   \
    _ONE_SPINOR_FOR_RED(s1, redop1, redop2)
 
#define _TWO_SITE_FOR_RED(s1, s2, redop1, redop2) \
    _CHECK_GEOMETRY_MATCHING(s1, s2);             \
    _ONE_SITE_FOR_RED(s1, redop1, redop2)
 
#define _TWO_SPINORS_FOR_SUM(s1, s2, ...) _TWO_SPINORS_FOR_RED(s1, s2, _omp_sum(__VA_ARGS__), )
 
#define _TWO_SITE_FOR_SUM(s1, s2, ...) _TWO_SITE_FOR_RED(s1, s2, _omp_sum(__VA_ARGS__), )
 
#define _THREE_SPINORS_FOR_RED(s1, s2, s3, redop1, redop2) \
    _TWO_SPINORS_MATCHING(s1, s2);                         \
    _TWO_SPINORS_MATCHING(s1, s3);                         \
    _ONE_SPINOR_FOR_RED(s1, redop1, redop2)
 
#define _THREE_SITE_FOR_RED(s1, s2, s3, redop1, redop2) \
    _CHECK_GEOMETRY_MATCHING(s1, s2);                   \
    _CHECK_GEOMETRY_MATCHING(s1, s3);                   \
    _ONE_SITE_FOR_RED(s1, redop1, redop2)
 
#define _ONE_SPINOR_FOR(s) _ONE_SPINOR_FOR_RED(s, , )
 
#define _ONE_SITE_FOR(s) _ONE_SITE_FOR_RED(s, , )
 
#define _TWO_SPINORS_FOR(s1, s2) _TWO_SPINORS_FOR_RED(s1, s2, , )
 
#define _TWO_SITE_FOR(s1, s2) _TWO_SITE_FOR_RED(s1, s2, , )
 
#define _THREE_SPINORS_FOR(s1, s2, s3) _THREE_SPINORS_FOR_RED(s1, s2, s3, , )
 
#define _THREE_SITE_FOR(s1, s2, s3) _THREE_SITE_FOR_RED(s1, s2, s3, , )
 
#define _SPINOR_PTR(s) _FIELD_AT(s, _spinor_for_is)
 
#define _SITE_PTR(__s, __mu, __dim) (_DFIELD_AT(__s, _site_for_is, __mu, __dim))
 
#endif