new_geometry.h

Go to the documentation of this file.

/***************************************************************************\
* Copyright (c) 2022, Claudio Pica, Sofie Martins                           *   
* All rights reserved.                                                      * 
\***************************************************************************/
 
 
#ifndef NEW_GEOMETRY_H
#define NEW_GEOMETRY_H
 
#include "spinor_field.h"
#include "suN.h"
#include "suN_types.h"
#include <stdint.h>
#include <stddef.h>
 
#ifdef __cplusplus
extern "C" {
#endif
 
void define_geometry(void);
void *sendbuf_alloc(size_t bytes_per_site);
#ifdef WITH_GPU
void *sendbuf_alloc_gpu(size_t bytes_per_site);
#endif
void sync_field(geometry_descriptor *gd, size_t byte_per_site, int is_spinor_like, void *latticebuf, void *sb_ptr);
int test_define_geometry(void);
void sendbuf_report(void);
void sync_field_gpu(geometry_descriptor *, int, int, void *, void *);
 
typedef struct coord4 {
    uint8_t x[4];
} coord4;
 
enum box_type {
    // L0 = 0, //unused
    // L1 = 1, //unused
    L2 = 2,
    L3 = 3,
    INNER = 4,
    SENDBUF = 5
};
 
typedef struct box_t {
    int l[4]; 
    int h[4]; 
    int base_index;
    int base_index_odd;
    int parity; 
    char mask; 
    enum box_type type; //< the type of the box, just a convenience for testing
    coord4 *icoord; 
    struct box_t *
        sendBox; 
    struct box_t *next; 
} box_t;
 
int boxEvenVolume(box_t *B);
int boxOddVolume(box_t *B);
int boxVolume(box_t *);
void geometryMemSize(box_t *G, size_t *total, size_t *buffers);
 
#define _DECLARE_SYNC_TO_BUFFER(_name, _field_type, _type) \
    void sync_box_to_buffer_gpu_##_name(geometry_descriptor *, box_t *, _field_type *, void *);
 
#define _DECLARE_SYNC_TO_BUFFER_REDUCED(_name, _field_type, _type)                                                 \
    void sync_box_to_buffer_gpu_reduced_##_name(geometry_descriptor *, box_t *, _field_type *, void *, int, char); \
    void sync_box_reduced_init_##_name(geometry_descriptor *, box_t *, _field_type *, void *, int);
 
_DECLARE_SYNC_TO_BUFFER(spinor_field, spinor_field, suNf_spinor);
_DECLARE_SYNC_TO_BUFFER(spinor_field_flt, spinor_field_flt, suNf_spinor_flt);
 
_DECLARE_SYNC_TO_BUFFER_REDUCED(spinor_field, spinor_field, suNf_spinor);
_DECLARE_SYNC_TO_BUFFER_REDUCED(spinor_field_flt, spinor_field_flt, suNf_spinor_flt)
 
_DECLARE_SYNC_TO_BUFFER(scalar_field, scalar_field, double);
 
_DECLARE_SYNC_TO_BUFFER(suNg_field, suNg_field, suNg);
_DECLARE_SYNC_TO_BUFFER(suNf_field, suNf_field, suNf);
_DECLARE_SYNC_TO_BUFFER(suNg_field_flt, suNg_field_flt, suNg_flt);
_DECLARE_SYNC_TO_BUFFER(suNf_field_flt, suNf_field_flt, suNf_flt);
_DECLARE_SYNC_TO_BUFFER(suNg_scalar_field, suNg_scalar_field, suNg_vector);
_DECLARE_SYNC_TO_BUFFER(suNg_av_field, suNg_av_field, suNg_algebra_vector);
_DECLARE_SYNC_TO_BUFFER(gtransf, gtransf, suNg);
_DECLARE_SYNC_TO_BUFFER(ldl_field, ldl_field, ldl_t);
_DECLARE_SYNC_TO_BUFFER(clover_term, clover_term, suNfc);
_DECLARE_SYNC_TO_BUFFER(clover_force, clover_force, suNf);
_DECLARE_SYNC_TO_BUFFER(staple_field, staple_field, suNg);
 
#ifdef __cplusplus
}
#endif
 
#endif