source: doc/theses/colby_parsons_MMAth/benchmarks/actors/ucpp/uC++Matrix.cc @ 55fabac

#include <iostream>
#include <algorithm>
using namespace std;
#include <chrono>
using namespace chrono;
#include <uActor.h>

unsigned int xr = 3'072, xc = 3'072, yc = 3'072, Processors = 1; // default values

struct WorkMsg : public uActor::Message {                               // derived message
        int * Z;
        const int * const X, * const * Y;
        WorkMsg( int Z[], const int X[], const int * const Y[] ) :
                Message( uActor::Finished ), Z( Z ), X( X ), Y( Y ) {} // one-shot
}; // WorkMsg

time_point<steady_clock> starttime;

_Actor MatrixMult {
        Allocation receive( Message & msg ) {
                Case ( WorkMsg, msg ) {
                        int * z = msg_d->Z;                                                     // optimizations
                        const int * const x = msg_d->X, * const * y = msg_d->Y;
                        for ( unsigned int i = 0; i < yc; i += 1 ) { // multiply X_row by Y_col and sum products
                                z[i] = 0;
                                for ( unsigned int j = 0; j < xc; j += 1 ) {
                                        z[i] += x[j] * y[j][i];
                                } // for
                        } // for
                } // Case

                return Finished;
        } // MatrixMult:::receive
}; // MatrixMult

int main( int argc, char * argv[] ) {
        locale loc( getenv("LANG") );
        cout.imbue( loc );

        switch ( argc ) {
          case 5:
                if ( strcmp( argv[4], "d" ) != 0 ) {                    // default ?
                        Processors = stoi( argv[4] );
                        if ( Processors < 1 ) goto Usage;
                } // if
          case 4:
                if ( strcmp( argv[3], "d" ) != 0 ) {                    // default ?
                        xr = stoi( argv[3] );
                        if ( xr < 1 ) goto Usage;
                } // if
          case 3:
                if ( strcmp( argv[2], "d" ) != 0 ) {                    // default ?
                        xc = stoi( argv[2] );
                        if ( xc < 1 ) goto Usage;
                } // if
          case 2:
                if ( strcmp( argv[1], "d" ) != 0 ) {                    // default ?
                        yc = stoi( argv[1] );
                        if ( yc < 1 ) goto Usage;
                } // if
          case 1:                                                                                       // use defaults
                break;
          default:
          Usage:
                cerr << "Usage: " << argv[0]
                         << " [ yc (> 0) | 'd' (default " << yc
                         << ") ] [ xc (> 0) | 'd' (default " << xc
                         << ") ] [ xr (> 0) | 'd' (default " << xr
                         << ") ] [ processors (> 0) | 'd' (default " << Processors
                         << ") ]" << endl;
                exit( EXIT_FAILURE );
        } // switch

        unsigned int r, c;
        int * Z[xr], * X[xr], * Y[xc];

        for ( r = 0; r < xr; r += 1 ) {                                         // create/initialize X matrix
                X[r] = new int[xc];
                for ( c = 0; c < xc; c += 1 ) {
                        X[r][c] = r * c % 37;                                           // for timing
                } // for
        } // for
        for ( r = 0; r < xc; r += 1 ) {                                         // create/initialize Y matrix
                Y[r] = new int[yc];
                for ( c = 0; c < yc; c += 1 ) {
                        Y[r][c] = r * c % 37;                                           // for timing
                } // for
        } // for
        for ( r = 0; r < xr; r += 1 ) {                                         // create Z matrix
                Z[r] = new int[yc];
        } // for

        uExecutor * executor = new uExecutor( Processors, Processors, Processors == 1 ? 1 : Processors * 32, true, 0 );
        uActor::start( executor );                                                      // start actor system
        uNoCtor<MatrixMult> * multiply = new uNoCtor<MatrixMult>[xr];
        uNoCtor<WorkMsg> * workMsg = new uNoCtor<WorkMsg>[xr];  

        for ( unsigned int r = 0; r < xr; r += 1 ) {
                multiply[r].ctor();
                workMsg[r].ctor( Z[r], X[r], (const int * const *)Y );
        } // for

        starttime = steady_clock::now();
        for ( unsigned int r = 0; r < xr; r += 1 ) {
                *multiply[r] | *workMsg[r];
        } // for

        uActor::stop();                                                                         // wait for all actors to terminate

        cout << (steady_clock::now() - starttime).count() / 1'000'000'000.0 << endl;
        
        for ( r = 0; r < xr; r += 1 ) {                                         // deallocate X and Z matrices
                delete [] X[r];
                delete [] Z[r];
        } // for
        for ( r = 0; r < xc; r += 1 ) {                                         // deallocate Y matrix
                delete [] Y[r];
        } // for

        // malloc_stats();
} // main

// /usr/bin/time -f "%Uu %Ss %Er %Mkb" a.out

// Local Variables: //
// compile-command: "u++-work -g -Wall -Wextra -O3 -nodebug -DNDEBUG -multi uC++Matrix.cc" //
// End: //