~gcrosswhite/charon/trunk

//@+leo-ver=4-thin

//@+node:gcross.20081204190733.4:@thin matmulspeed.C

//@@language cplusplus

 

#include "matmulspeed.h"

#include "matrix_multiplication.h"

 

matmulspeed::matmulspeed(CkArgMsg *m) {

    CProxy_matmulspeed self(thishandle);

    self.runTest();

}

 

void doTest(int I, int J, int K) {

    typedef Array<double,2> Matrix;

 

    CProxy_MatrixMultiplier<double> multiplier = CProxy_MatrixMultiplier<double>::ckNew(I,J,K);

    CkGroupID mCastGrpId = CProxy_CkMulticastMgr::ckNew();

    Array<CProxySection_MatrixMultiplier<double>,2> A_section_matrix(I,J), B_section_matrix(J,K), C_section_matrix(I,K);

    MatrixMultiplier<double>::build_section_matrices(

        multiplier,

        I,J,K,

        mCastGrpId,

        A_section_matrix,

        B_section_matrix,

        C_section_matrix

    );

    Array<CkCallback,2> C_callback_matrix(I,K);

    MatrixMultiplier<double>::build_thread_delay_callback_matrix(mCastGrpId,C_section_matrix,C_callback_matrix);

 

    {

        Matrix A_data(23,11);

        Matrix B_data(11,13);

        Matrix C_data(23,13), correct_C_data(23,13);

 

        cout << "Loading matrix A..." << endl;

        //@        << A data >>

        //@+node:gcross.20081204190733.12:<< A data >>

        //@-node:gcross.20081204190733.12:<< A data >>

        //@nl

        cout << "Loading matrix B..." << endl;

        //@        << B data >>

        //@+node:gcross.20081204190733.13:<< B data >>

        //@-node:gcross.20081204190733.13:<< B data >>

        //@nl

        cout << "Loading matrix C..." << endl;

        //@        << C data >>

        //@+node:gcross.20081204190733.14:<< C data >>

        //@-node:gcross.20081204190733.14:<< C data >>

        //@nl

 

        MatrixMultiplier<double>::broadcast_A_data(0,A_section_matrix,A_data);

        MatrixMultiplier<double>::broadcast_B_data(0,B_section_matrix,B_data);

        MatrixMultiplier<double>::gather_results(C_callback_matrix,C_data);

        assert(sum(abs(C_data-correct_C_data))<1e-8);

        cout << "They agree!" << endl;

    }

 

    multiplier.ckDestroy();

}

 

#include <ctime>

 

void matmulspeed::runTest() {

 

    clock_t timestamp = clock();

    doTest(1,1,1);

    cout << (clock()-timestamp) << endl;

 

    CkExitAfterQuiescence();

 

}

 

#include "matmulspeed.def.h"

 

//@-node:gcross.20081204190733.4:@thin matmulspeed.C

//@-leo