/var/lib/jenkins/jobs/Skylark/workspace/sketch/capi/sketchc.hpp

Go to the documentation of this file.

#ifndef SKYLARK_SKETCHC_HPP
#define SKYLARK_SKETCHC_HPP

#include "mpi.h"
#include "../../config.h"
#include "../../utility/distributions.hpp"

#include "../sketch.hpp"
#include "../../base/context.hpp"


// Some tools require special API declaration. Customizing the
// SKYLARK_EXTERN_API allows this. The default is simply nothing.
#ifndef SKYLARK_EXTERN_API
#define SKYLARK_EXTERN_API
#endif

namespace skylark {
namespace sketch {
namespace c {

enum transform_type_t {
    TRANSFORM_TYPE_ERROR,
    JLT,
    CT,
    FJLT,
    CWT,
    MMT,
    WZT,
    PPT,
    GaussianRFT,
    LaplacianRFT,
    FastGaussianRFT,
    ExpSemigroupRLT
};

enum matrix_type_t {
    MATRIX_TYPE_ERROR,
    MATRIX,                     
    SHARED_MATRIX,              
    ROOT_MATRIX,                
    DIST_MATRIX,                
    DIST_MATRIX_VC_STAR,
    DIST_MATRIX_VR_STAR,
    DIST_MATRIX_STAR_VC,
    DIST_MATRIX_STAR_VR,
    DIST_SPARSE_MATRIX,          
    SPARSE_MATRIX                
};

struct sketch_transform_t {
    const transform_type_t type;
    sketch_transform_data_t * const transform_obj;

    sketch_transform_t(transform_type_t type,
        sketch_transform_data_t *transform_obj)
        : type(type), transform_obj(transform_obj) {}
};

} // namespace c
} // namespace sketch
} // namespace skylark

namespace base    = skylark::base;
namespace sketch  = skylark::sketch;
namespace sketchc = skylark::sketch::c;

#define SL_COLUMNWISE 1
#define SL_ROWWISE    2

extern "C" {

SKYLARK_EXTERN_API char *sl_supported_sketch_transforms();

SKYLARK_EXTERN_API const char* sl_strerror(const int errorcode);

SKYLARK_EXTERN_API bool sl_has_elemental();
SKYLARK_EXTERN_API bool sl_has_combblas();


// Support for skylark::base::context_t.

SKYLARK_EXTERN_API int sl_create_default_context(int seed,
                                                 base::context_t **ctxt);

SKYLARK_EXTERN_API int sl_create_context(int seed, MPI_Comm cm,
                                         base::context_t **ctxt);

SKYLARK_EXTERN_API int sl_free_context(base::context_t *ctxt);

SKYLARK_EXTERN_API int sl_create_sketch_transform(
        base::context_t *ctxt, char *type,
        int n, int s, sketchc::sketch_transform_t **sketch, ...);

SKYLARK_EXTERN_API int sl_deserialize_sketch_transform(
       const char *data, sketchc::sketch_transform_t **sketch);

SKYLARK_EXTERN_API int sl_serialize_sketch_transform(
       const sketchc::sketch_transform_t *sketch, char **data);

SKYLARK_EXTERN_API int sl_free_sketch_transform(
        sketchc::sketch_transform_t *S);

SKYLARK_EXTERN_API int sl_apply_sketch_transform(
        sketchc::sketch_transform_t *S,
        char *input_type, void *A,
        char *output_type, void *SA, int dim);

// Helper functions to allow wrapping of object

SKYLARK_EXTERN_API int sl_wrap_raw_matrix(double *data, int m, int n, void **A);

SKYLARK_EXTERN_API int sl_free_raw_matrix_wrap(void *A_);

SKYLARK_EXTERN_API int sl_wrap_raw_sp_matrix(int *indptr, int *ind,
    double *data, int nnz, int n_rows, int n_cols, void **A);

SKYLARK_EXTERN_API int sl_free_raw_sp_matrix_wrap(void *A_);

SKYLARK_EXTERN_API int sl_raw_sp_matrix_struct_updated(void *A_,
    bool *struct_updated);

SKYLARK_EXTERN_API int sl_raw_sp_matrix_reset_update_flag(void *A_);

SKYLARK_EXTERN_API int sl_raw_sp_matrix_nnz(void *A_, int *nnz);

SKYLARK_EXTERN_API int sl_raw_sp_matrix_data(void *A_, int32_t *indptr,
        int32_t *indices, double *values);

} // extern "C"

#endif // SKYLARK_SKETCHC_HPP