api_matmul.cpp

#include <iostream>
#include <fstream>
#include <vector>
#include <cstdlib>
#include <unistd.h>

#include "../src/Mitos.h"
#include "src/virtual_address_writer.h"

mitos_output mout;

void sample_handler(perf_event_sample *sample, void *args)
{
    //fprintf(stderr, "SAMP: cpu=%d, tid=%d\n", sample->cpu, sample->tid);
    Mitos_write_sample(sample, &mout);
}

#define ROW_MAJOR(x,y,width) y*width+x

void init_matrices(int N, double **a, double **b, double **c)
{
    int i,j,k;

    *a = new double[N*N];
    *b = new double[N*N];
    *c = new double[N*N];

    size_t dims[2];
    dims[0] = N; //static_cast<size_t>(N);
    dims[1] = N; //static_cast<size_t>(N);
    Mitos_add_symbol("a",*a,sizeof(double),dims,2);
    Mitos_add_symbol("b",*b,sizeof(double),dims,2);
    Mitos_add_symbol("c",*c,sizeof(double),dims,2);

    for(i=0; i<N; ++i)
    {
        for(j=0; j<N; ++j)
        {
            (*a)[ROW_MAJOR(i,j,N)] = (double)rand();
            (*b)[ROW_MAJOR(i,j,N)] = (double)rand();
            (*c)[ROW_MAJOR(i,j,N)] = 0;
        }
    }
}

void matmul(int N, double *a, double *b, double *c)
{
    for(int i=0; i<N; ++i)
    {
        for(int j=0; j<N; ++j)
        {
            for(int k=0; k<N; ++k)
            {
                c[ROW_MAJOR(i,j,N)] += a[ROW_MAJOR(i,k,N)]*b[ROW_MAJOR(k,j,N)];
            }
        }
    }

    int randx = N*((float)rand() / (float)RAND_MAX+1);
    int randy = N*((float)rand() / (float)RAND_MAX+1);
    std::cout << c[ROW_MAJOR(randx,randy,N)] << std::endl;
}

int main(int argc, char **argv)
{
    int N = (argc == 2) ? atoi(argv[1]) : 1024;

    double *a,*b,*c;
    init_matrices(N,&a,&b,&c);

    auto pid = getpid();

    auto unique_id = time(NULL);
    // Create output directories and get the location of the virtual address file to be created
    Mitos_create_output(&mout, unique_id);
    
    std::string virt_address = "/tmp/mitos_virt_addres.txt";
    Mitos_save_virtual_address_offset(virt_address);
    Mitos_pre_process(&mout);

    Mitos_set_pid(pid);
    Mitos_set_handler_fn(&sample_handler,NULL);
    Mitos_set_sample_latency_threshold(3);
    Mitos_set_sample_event_period(4000);

    std::cout << "[Mitos] Beginning sampler\n";
    Mitos_begin_sampler();
    matmul(N,a,b,c);
    Mitos_end_sampler();
    std::cout << "[Mitos] End sampler\n";
    Mitos_add_offsets(virt_address.c_str(), &mout);
    if(Mitos_process_binary(argv[0], &mout))
    {
        std::cerr << "Error opening binary file!" << std::endl;
        return 1;
    }
    if(Mitos_post_process(argv[0],&mout, mout.dname_topdir)){
        std::cerr << "Error post processing!" << std::endl;
        return 1;
    }
    return 0;
}