gensvm_zv.c

Go to the documentation of this file.

#include "gensvm_zv.h"

void gensvm_calculate_ZV(struct GenModel *model, struct GenData *data,
        double *ZV)
{
    if (data->Z == NULL)
        gensvm_calculate_ZV_sparse(model, data, ZV);
    else
        gensvm_calculate_ZV_dense(model, data, ZV);
}

void gensvm_calculate_ZV_sparse(struct GenModel *model,
        struct GenData *data, double *ZV)
{
    long i, j, jj, jj_start, jj_end, K,
        n_row = data->spZ->n_row;
    double z_ij;

    K = model->K;

    long *Zia = data->spZ->ia;
    long *Zja = data->spZ->ja;
    double *vals = data->spZ->values;

    for (i=0; i<n_row; i++) {
        jj_start = Zia[i];
        jj_end = Zia[i+1];

        for (jj=jj_start; jj<jj_end; jj++) {
            j = Zja[jj];
            z_ij = vals[jj];

            cblas_daxpy(K-1, z_ij, &model->V[j*(K-1)], 1,
                    &ZV[i*(K-1)], 1);
        }
    }
}

void gensvm_calculate_ZV_dense(struct GenModel *model,
        struct GenData *data, double *ZV)
{
    // use n from data, assume m and K are the same between model and data
    long n = data->n;
    long m = model->m;
    long K = model->K;

    cblas_dgemm(CblasRowMajor, CblasNoTrans, CblasNoTrans, n, K-1, m+1,
            1.0, data->Z, m+1, model->V, K-1, 0, ZV, K-1);
}