distr_matrix.h

Go to the documentation of this file.

#ifndef STARNEIG_DISTR_MATRIX_H
#define STARNEIG_DISTR_MATRIX_H

#ifdef __cplusplus
extern "C" {
#endif

#include <starneig/configuration.h>

#ifndef STARNEIG_ENABLE_MPI
#error "This should be included only when STARNEIG_ENABLE_MPI is defined."
#endif

#include <stddef.h>



struct starneig_distr;

typedef struct starneig_distr * starneig_distr_t;

typedef enum {
    STARNEIG_ORDER_DEFAULT,     
    STARNEIG_ORDER_ROW_MAJOR,   
    STARNEIG_ORDER_COL_MAJOR    
} starneig_distr_order_t;

starneig_distr_t starneig_distr_init();

starneig_distr_t starneig_distr_init_mesh(
    int rows, int cols, starneig_distr_order_t order);


starneig_distr_t starneig_distr_init_func(
    int (*func)(int row, int col, void *arg), void *arg, size_t arg_size);

starneig_distr_t starneig_distr_duplicate(starneig_distr_t distr);

void starneig_distr_destroy(starneig_distr_t distr);



struct starneig_distr_matrix;

typedef struct starneig_distr_matrix * starneig_distr_matrix_t;

typedef enum {
    STARNEIG_REAL_DOUBLE  
} starneig_datatype_t;

struct starneig_distr_block {
    int row_blksz;  
    int col_blksz;  
    int glo_row;    
    int glo_col;    
    int ld;         
    void *ptr;      
};

starneig_distr_matrix_t starneig_distr_matrix_create(
    int rows, int cols, int row_blksz, int col_blksz, starneig_datatype_t type,
    starneig_distr_t distr);

starneig_distr_matrix_t starneig_distr_matrix_create_local(
    int rows, int cols, starneig_datatype_t type, int owner, double *A,
    int ldA);

void starneig_distr_matrix_destroy(starneig_distr_matrix_t matrix);

void starneig_distr_matrix_copy(
    starneig_distr_matrix_t source, starneig_distr_matrix_t dest);

void starneig_distr_matrix_copy_region(
    int sr, int sc, int dr, int dc, int rows, int cols,
    starneig_distr_matrix_t source, starneig_distr_matrix_t dest);



void starneig_distr_matrix_get_blocks(
    starneig_distr_matrix_t matrix, struct starneig_distr_block **blocks,
    int *num_blocks);

starneig_distr_t starneig_distr_matrix_get_distr(
    starneig_distr_matrix_t matrix);

starneig_datatype_t starneig_distr_matrix_get_datatype(
    starneig_distr_matrix_t matrix);

size_t starneig_distr_matrix_get_elemsize(starneig_distr_matrix_t matrix);

int starneig_distr_matrix_get_rows(starneig_distr_matrix_t matrix);

int starneig_distr_matrix_get_cols(starneig_distr_matrix_t matrix);

int starneig_distr_matrix_get_row_blksz(starneig_distr_matrix_t matrix);

int starneig_distr_matrix_get_col_blksz(starneig_distr_matrix_t matrix);



void starneig_broadcast(int root, size_t size, void *buffer);



#ifdef __cplusplus
}
#endif

#endif // STARNEIG_DISTR_MATRIX_H