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 header 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);
 
 
 
// deprecated
void starneig_broadcast(int root, size_t size, void *buffer);
 
#ifdef __cplusplus
}
#endif
 
#endif // STARNEIG_DISTR_MATRIX_H