starsh-randtlr.h

Go to the documentation of this file.

#ifndef __STARSH_RANDTLR_H__
#define __STARSH_RANDTLR_H__

// Add definitions for size_t, va_list and STARSH_kernel
#include "starsh.h"

typedef struct starsh_randtlr

{
    STARSH_int count;
    //char dtype; // not supported yet
    STARSH_int nblocks;
    STARSH_int block_size;
    double *U;
    double *S;
    double diag;
} STARSH_randtlr;

enum STARSH_RANDTLR_KERNEL

{
    STARSH_RANDTLR_KERNEL1 = 1,
};

enum STARSH_RANDTLR_PARAM

{
    STARSH_RANDTLR_NB = 1,
    STARSH_RANDTLR_DECAY = 2,
    STARSH_RANDTLR_DIAG = 3,
};

int starsh_randtlr_generate(STARSH_randtlr **data, STARSH_int count,
        STARSH_int block_size, double decay, double diag);
int starsh_randtlr_generate_va(STARSH_randtlr **data, STARSH_int count,
        va_list args);
int starsh_randtlr_generate_el(STARSH_randtlr **data, STARSH_int count, ...);
int starsh_randtlr_get_kernel(STARSH_kernel **kernel, STARSH_randtlr *data,
        enum STARSH_RANDTLR_KERNEL type);
void starsh_randtlr_free(STARSH_randtlr *data);

// KERNELS
void starsh_randtlr_block_kernel(int nrows, int ncols, STARSH_int *irow,
        STARSH_int *icol, void *row_data, void *col_data, void *result,
        int ld);

#endif // __STARSH_RANDTLR_H__