spinor_field.h

Go to the documentation of this file.

/***************************************************************************\
* Copyright (c) 2008, 2022, Claudio Pica, Sofie Martins                     *   
* All rights reserved.                                                      * 
\***************************************************************************/
 
#ifndef SPINOR_FIELD_H
#define SPINOR_FIELD_H
 
#include "Geometry/geometry_descriptor.h"
#include "suN_types.h"
#ifdef WITH_MPI
#include "hr_mpi.h"
#endif
 
/* Communications */
typedef enum comm_t { CPU_COMM = 1 << 0, GPU_COMM = 1 << 1 } comm_t;
#define ALL_COMMS (CPU_COMM | GPU_COMM)
 
/* Memory */
typedef enum mem_t { CPU_MEM = 1 << 0, GPU_MEM = 1 << 1 } mem_t;
 
#ifdef __cplusplus
extern "C" {
#endif
 
#define _GPU_FIELD_DATA(_type)
#ifdef WITH_GPU
#undef _GPU_FIELD_DATA
#define _GPU_FIELD_DATA(_type) _type *gpu_ptr;
#endif
 
#define _MPI_FIELD_DATA(_type)
#ifdef WITH_MPI
#undef _MPI_FIELD_DATA
#define _MPI_FIELD_DATA(_type) \
    MPI_Request *comm_req;     \
    _type *sendbuf_ptr;        \
    _type *sendbuf_gpu_ptr;    \
    _type *recvbuf_gpu_ptr;
 
//
#endif
 
#define _DECLARE_FIELD_STRUCT(_name, _type) \
    typedef struct _name {                  \
        _type *ptr;                         \
        geometry_descriptor *type;          \
        int field_dim;                      \
        _MPI_FIELD_DATA(_type)              \
        _GPU_FIELD_DATA(_type)              \
        comm_t comm_type;                   \
        mem_t alloc_t;                      \
    } _name
 
_DECLARE_FIELD_STRUCT(suNg_field, suNg);
 
_DECLARE_FIELD_STRUCT(suNg_scalar_field, suNg_vector);
 
_DECLARE_FIELD_STRUCT(suNg_field_flt, suNg_flt);
 
_DECLARE_FIELD_STRUCT(suNf_field, suNf);
 
_DECLARE_FIELD_STRUCT(suNf_field_flt, suNf_flt);
 
_DECLARE_FIELD_STRUCT(spinor_field, suNf_spinor);
 
_DECLARE_FIELD_STRUCT(spinor_field_flt, suNf_spinor_flt);
 
_DECLARE_FIELD_STRUCT(suNg_av_field, suNg_algebra_vector);
 
_DECLARE_FIELD_STRUCT(scalar_field, double);
 
_DECLARE_FIELD_STRUCT(ldl_field, ldl_t);
 
_DECLARE_FIELD_STRUCT(suNfc_field, suNfc);
 
_DECLARE_FIELD_STRUCT(clover_term, suNfc);
 
_DECLARE_FIELD_STRUCT(clover_force, suNf);
 
_DECLARE_FIELD_STRUCT(staple_field, suNg);
 
_DECLARE_FIELD_STRUCT(gtransf, suNg);
 
#define _FIELD_TYPE spinor_field
#define _FIELD_DIM 1
#include "TMPL/spinor_field.h.tmpl"
 
#define _FIELD_TYPE spinor_field_flt
#define _FIELD_DIM 1
#include "TMPL/spinor_field.h.tmpl"
 
#define _FIELD_TYPE scalar_field
#define _FIELD_DIM 1
#include "TMPL/spinor_field.h.tmpl"
 
#define _FIELD_TYPE suNg_field
#define _FIELD_DIM 4
#include "TMPL/spinor_field.h.tmpl"
 
#define _FIELD_TYPE suNg_field_flt
#define _FIELD_DIM 4
#include "TMPL/spinor_field.h.tmpl"
 
#define _FIELD_TYPE suNf_field
#define _FIELD_DIM 4
#include "TMPL/spinor_field.h.tmpl"
 
#define _FIELD_TYPE suNf_field_flt
#define _FIELD_DIM 4
#include "TMPL/spinor_field.h.tmpl"
 
#define _FIELD_TYPE suNg_scalar_field
#define _FIELD_DIM 1
#include "TMPL/spinor_field.h.tmpl"
 
#define _FIELD_TYPE suNg_av_field
#define _FIELD_DIM 4
#include "TMPL/spinor_field.h.tmpl"
 
#define _FIELD_TYPE gtransf
#define _FIELD_DIM 1
#include "TMPL/spinor_field.h.tmpl"
 
#define _FIELD_TYPE ldl_field
#define _FIELD_DIM 1
#include "TMPL/spinor_field.h.tmpl"
 
#define _FIELD_TYPE clover_term
#define _FIELD_DIM 4
#include "TMPL/spinor_field.h.tmpl"
 
#define _FIELD_TYPE clover_force
#define _FIELD_DIM 6
#include "TMPL/spinor_field.h.tmpl"
 
#define _FIELD_TYPE staple_field
#define _FIELD_DIM 3
#include "TMPL/spinor_field.h.tmpl"
 
#ifdef __cplusplus
}
#endif
#endif