source: soft/giet_vm/transpose/main.c @ 333

///////////////////////////////////////////////////////////////////////////////////////////////
// File   : main.c   (for transpose application)
// Date   : february 2014
// author : Alain Greiner
//
// This application makes a transpose for a NN*NN pixels sequence of images.
// The image sequence is read from a file (one byte per pixel).
// The input and output buffers containing the image are distributed in all clusters.
//
// - The image size NN must be a power of 2.
// - The number of clusters containing processors must be a power of 2.
// - The number of processors per cluster (NB_PROCS_MAX) must be a power of 2.
// - The image size NN must be larger or equal to the total number of processor.
//
// For each image the application makes a self test (checksum for each line).
// The actual display on the frame buffer is optional (controled by DISPLAY_OK define)
///////////////////////////////////////////////////////////////////////////////////////////////

#include "hard_config.h"
#include "stdio.h"
#include "malloc.h"
#include "barrier.h"

#define NN                  128                 // image size : nlines = npixels = 128
#define NB_IMAGES           5                   // number of images to be handled
#define FILE_PATHNAME       "misc/images.raw"   // file pathname on disk

#define INSTRUMENTATION_OK  0                   // display statistcs on TTY when non zero

///////////////////////////////////////////////////////
// global variables stored in seg_data in cluster(0,0)
///////////////////////////////////////////////////////

// instrumentation counters 
// for each processor (up to 4 processors)
// in each cluster (up to 32 clusters)
unsigned int LOAD_START[32][4];
unsigned int LOAD_END  [32][4];
unsigned int TRSP_START[32][4];
unsigned int TRSP_END  [32][4];
unsigned int DISP_START[32][4];
unsigned int DISP_END  [32][4];

// arrays of pointers on distributed buffers
// one input buffer & one output buffer per cluster
unsigned char*  buf_in [32];
unsigned char*  buf_out[32];

// checksum variables 
unsigned check_line_before[NN];
unsigned check_line_after[NN];

// synchronisation barriers
giet_barrier_t barrier_0;
giet_barrier_t barrier_1;
giet_barrier_t barrier_2;
giet_barrier_t barrier_3;
giet_barrier_t barrier_4;

volatile unsigned int init_ok = 1;

//////////////////////////////////////////
__attribute__ ((constructor)) void main()
{

    int          file;                                             // file descriptor for images
    unsigned int l;                                                // line index for loops
    unsigned int p;                                                // pixel index for loops

    unsigned int proc_id     = giet_procid();                      // processor id
    unsigned int lpid        = proc_id % NB_PROCS_MAX;             // local processor id
    unsigned int cluster_xy  = proc_id / NB_PROCS_MAX;             // cluster index (8 bits format)
    unsigned int x           = cluster_xy >> Y_WIDTH;              // x coordinate
    unsigned int y           = cluster_xy & ((1<<Y_WIDTH)-1);      // y coordinate

    unsigned int npixels     = NN * NN;                            // number of pixel per image
    unsigned int nblocks     = npixels / 512;                      // number of blocks per image
    unsigned int image       = 0;                                  // image counter

    unsigned int nclusters   = X_SIZE * Y_SIZE;                    // number of clusters with procs
    unsigned int cluster_id  = (x * Y_SIZE) + y;                   // "continuous" cluster index   
    unsigned int ntasks      = nclusters * NB_PROCS_MAX;           // number of tasks
    unsigned int task_id     = (cluster_id * NB_PROCS_MAX) +lpid;  // "continuous" task index

    // processor [0,0,0] makes parameters checking and barriers initialization
    if ( proc_id == 0 )
    {
        if ((NB_PROCS_MAX != 1) && (NB_PROCS_MAX != 2) && (NB_PROCS_MAX != 4))
        { 
            giet_exit("[TRANSPOSE ERROR] NB_PROCS_MAX must be 1, 2 or 4");
        }
        if ((nclusters != 1) && (nclusters != 2) && (nclusters != 4) && (nclusters != 8) &&
            (nclusters != 16) && (nclusters != 32) )
        {
            giet_exit("[TRANSPOSE ERROR] number of clusters must be 2,4,8,16,32");
        }
        if ( ntasks > NN )
        {
            giet_exit("[TRANSPOSE ERROR] number of tasks larger than number of lines");
        }

        if ( ntasks > 1 )
        {
            barrier_init( &barrier_0, ntasks );
            barrier_init( &barrier_1, ntasks );
            barrier_init( &barrier_2, ntasks );
            barrier_init( &barrier_3, ntasks );
            barrier_init( &barrier_4, ntasks );
        }

        giet_shr_printf("\n[TRANSPOSE] Proc [%d,%d,%d] completes barrier init at cycle %d\n",
                        0, 0, 0, giet_proctime() );
        giet_shr_printf(" - nclusters = %d\n", nclusters );
        giet_shr_printf(" - ntasks    = %d\n", ntasks );

        init_ok = 0;
    }
    else   // others processors wait initialisation completion
    {
        while ( init_ok == 1 );
    }
    
    // The buffers containing the images are distributed in clusters
    // (one buf_in and one buf_out per cluster).
    // Each buffer contains (NN*NN / nclusters) bytes.
    // They are allocated in the cluster[x,y] heap by processor[x,y,0]
    if ( lpid == 0 )
    {
        // get heap vaddr in cluster[0,0]
        unsigned int heap_base;         
        giet_vobj_get_vbase( "transpose", "trsp_heap_0_0", &heap_base );
 
        // allocate buffers in cluster[x,y]
        buf_in[cluster_id]  = ((unsigned char*)heap_base) + (cluster_xy << 20);
        buf_out[cluster_id] = buf_in[cluster_id] + NN*NN/nclusters;

        giet_shr_printf("\n[TRANSPOSE] Proc [%d,%d,%d] completes buffer allocation at cycle %d\n"
                        " - buf_in  = %x\n"
                        " - buf_out = %x\n",
                        x, y, 0, giet_proctime(), 
                        (unsigned int)buf_in[cluster_id], 
                        (unsigned int)buf_out[cluster_id] );

        // open file containing images
        file = giet_fat_open( "misc/images.raw", 0);

        if (file < 0)
        { 
            giet_shr_printf("[TRANSPOSE ERROR] Processor[%d,%d,%d]"
                            " cannot open file misc/images.raw",
                            x, y, lpid );
            giet_exit(" open() failure");
        }
        else
        {
            giet_shr_printf("\n[TRANSPOSE] Processor[%d,%d,%d]"
                            " open file misc/images.raw\n",
                            x, y, lpid );
        }
    }

    /////////////////////////////////////////////
    if ( ntasks > 1 ) barrier_wait( &barrier_0 );

    // Main loop (on images)
    while (image < NB_IMAGES)
    {
        // pseudo parallel load from disk to buf_in buffer : nblocks/nclusters blocks
        // only task running on processor with (lpid == 0) does it

        LOAD_START[cluster_id][lpid] = giet_proctime();

        if (lpid == 0)
        {
            giet_fat_read( file,
                           buf_in[cluster_id],
                           (nblocks / nclusters),
                           ((image*nblocks) + ((nblocks*cluster_id)/nclusters)) );

            giet_shr_printf("\n[TRANSPOSE] Proc [%d,%d,0] completes load"
                            "  for image %d at cycle %d\n",
                            x, y, image, giet_proctime() );
        }

        LOAD_END[cluster_id][lpid] = giet_proctime();

        /////////////////////////////////////////////
        if ( ntasks > 1 ) barrier_wait( &barrier_1 );

        // parallel transpose from buf_in to buf_out
        // each task makes the transposition for nlt lines (nlt = NN/ntasks)
        // from line [task_id*nlt] to line [(task_id + 1)*nlt - 1]
        // (p,l) are the absolute pixel coordinates in the source image


        TRSP_START[cluster_id][lpid] = giet_proctime();

        unsigned int nlt   = NN / ntasks;      // number of lines per task
        unsigned int nlc   = NN / nclusters;   // number of lines per cluster

        unsigned int src_cluster;
        unsigned int src_index;
        unsigned int dst_cluster;
        unsigned int dst_index;

        unsigned char byte;

        unsigned int first = task_id * nlt;    // first line index for a given task
        unsigned int last  = first + nlt;      // last line index for a given task

        for ( l = first ; l < last ; l++ )
        {
            check_line_before[l] = 0;
         
            // in each iteration we transfer one byte
            for ( p = 0 ; p < NN ; p++ )
            {
                // read one byte from local buf_in
                src_cluster = l / nlc;
                src_index   = (l % nlc)*NN + p;
                byte        = buf_in[src_cluster][src_index];

                // compute checksum
                check_line_before[l] = check_line_before[l] + byte;

                // write one byte to remote buf_out
                dst_cluster = p / nlc; 
                dst_index   = (p % nlc)*NN + l;
                buf_out[dst_cluster][dst_index] = byte;
            }
        }

        if ( lpid == 0 )
        {
            giet_shr_printf("\n[TRANSPOSE] proc [%d,%d,0] completes transpose"
                            " for image %d at cycle %d\n", 
                            x, y, image, giet_proctime() );

        }
        TRSP_END[cluster_id][lpid] = giet_proctime();

        /////////////////////////////////////////////
        if ( ntasks > 1 ) barrier_wait( &barrier_2 );

        // optional parallel display from local buf_out to frame buffer
        // all processors contribute to display using memcpy...

        if ( USE_FBF )  // external frame buffer available
        {
            DISP_START[cluster_id][lpid] = giet_proctime();

            unsigned int  npt   = npixels / ntasks;   // number of pixels per task

            giet_fb_sync_write( npt * task_id, 
                                &buf_out[cluster_id][lpid*npt], 
                                npt );

            if ( lpid == 0 )
            {
                giet_shr_printf("\n[TRANSPOSE] Proc [%d,%d,0] completes display"
                                " for image %d at cycle %d\n",
                                x, y, image, giet_proctime() );
            }

            DISP_END[cluster_id][lpid] = giet_proctime();

            /////////////////////////////////////////////
            if ( ntasks > 1 ) barrier_wait( &barrier_3 );
        }

        // checksum done by processor (lpid == 0) in each cluster

        if ( lpid == 0 )
        {
            unsigned int success = 1;
            unsigned int start   = cluster_id * nlc;
            unsigned int stop    = start + nlc;

            for ( l = start ; l < stop ; l++ )
            {
                check_line_after[l] = 0;

                for ( p = 0 ; p < NN ; p++ )
                {
                    // read one byte in remote buffer
                    src_cluster = p / nlc;
                    src_index   = (p % nlc)*NN + l;

                    unsigned char byte = buf_out[src_cluster][src_index];

                    check_line_after[l] = check_line_after[l] + byte;
                }

                if ( check_line_before[l] != check_line_after[l] ) success = 0;
            }

            if ( success ) 
            {
                giet_shr_printf("\n[TRANSPOSE] proc [%d,%d,0] checksum OK"
                                " for image %d\n",
                                x, y, image );
            }
            else
            {
                giet_shr_printf("\n[TRANSPOSE] proc [%d,%d,0] checksum KO"
                                " for image %d\n",
                                x, y, image );
            }
        }

        // instrumentation done by processor [0,0,0]

        if ( (proc_id == 0) && INSTRUMENTATION_OK )
        {
            int cc, pp;
            unsigned int min_load_start = 0xFFFFFFFF;
            unsigned int max_load_start = 0;
            unsigned int min_load_ended = 0xFFFFFFFF;
            unsigned int max_load_ended = 0;
            unsigned int min_trsp_start = 0xFFFFFFFF;
            unsigned int max_trsp_start = 0;
            unsigned int min_trsp_ended = 0xFFFFFFFF;
            unsigned int max_trsp_ended = 0;
            unsigned int min_disp_start = 0xFFFFFFFF;
            unsigned int max_disp_start = 0;
            unsigned int min_disp_ended = 0xFFFFFFFF;
            unsigned int max_disp_ended = 0;

            for (cc = 0; cc < nclusters; cc++)
            {
                for (pp = 0; pp < NB_PROCS_MAX; pp++)
                {
                    if (LOAD_START[cc][pp] < min_load_start)  min_load_start = LOAD_START[cc][pp];
                    if (LOAD_START[cc][pp] > max_load_start)  max_load_start = LOAD_START[cc][pp];
                    if (LOAD_END[cc][pp]   < min_load_ended)  min_load_ended = LOAD_END[cc][pp]; 
                    if (LOAD_END[cc][pp]   > max_load_ended)  max_load_ended = LOAD_END[cc][pp];
                    if (TRSP_START[cc][pp] < min_trsp_start)  min_trsp_start = TRSP_START[cc][pp];
                    if (TRSP_START[cc][pp] > max_trsp_start)  max_trsp_start = TRSP_START[cc][pp];
                    if (TRSP_END[cc][pp]   < min_trsp_ended)  min_trsp_ended = TRSP_END[cc][pp];
                    if (TRSP_END[cc][pp]   > max_trsp_ended)  max_trsp_ended = TRSP_END[cc][pp];
                    if (DISP_START[cc][pp] < min_disp_start)  min_disp_start = DISP_START[cc][pp];
                    if (DISP_START[cc][pp] > max_disp_start)  max_disp_start = DISP_START[cc][pp];
                    if (DISP_END[cc][pp]   < min_disp_ended)  min_disp_ended = DISP_END[cc][pp];
                    if (DISP_END[cc][pp]   > max_disp_ended)  max_disp_ended = DISP_END[cc][pp];
                }
            }

            giet_shr_printf(" - LOAD_START : min = %d / max = %d / med = %d / delta = %d\n",
                            min_load_start, max_load_start, (min_load_start+max_load_start)/2, 
                            max_load_start-min_load_start); 

            giet_shr_printf(" - LOAD_END   : min = %d / max = %d / med = %d / delta = %d\n",
                            min_load_ended, max_load_ended, (min_load_ended+max_load_ended)/2, 
                            max_load_ended-min_load_ended); 

            giet_shr_printf(" - TRSP_START : min = %d / max = %d / med = %d / delta = %d\n",
                            min_trsp_start, max_trsp_start, (min_trsp_start+max_trsp_start)/2, 
                            max_trsp_start-min_trsp_start); 

            giet_shr_printf(" - TRSP_END   : min = %d / max = %d / med = %d / delta = %d\n",
                            min_trsp_ended, max_trsp_ended, (min_trsp_ended+max_trsp_ended)/2, 
                            max_trsp_ended-min_trsp_ended); 

            giet_shr_printf(" - DISP_START : min = %d / max = %d / med = %d / delta = %d\n",
                            min_disp_start, max_disp_start, (min_disp_start+max_disp_start)/2, 
                            max_disp_start-min_disp_start); 

            giet_shr_printf(" - DISP_END   : min = %d / max = %d / med = %d / delta = %d\n",
                            min_disp_ended, max_disp_ended, (min_disp_ended+max_disp_ended)/2, 
                            max_disp_ended-min_disp_ended); 
        }

        image++;

        //////////////////////////////////////////////////
        // all tasks wait instrumentation completion 
        //////////////////////////////////////////////////
        if ( ntasks > 1 ) barrier_wait( &barrier_4 );

    } // end while image      

    giet_exit("Completed");

} // end main()

// Local Variables:
// tab-width: 3
// c-basic-offset: 
// c-file-offsets:((innamespace . 0)(inline-open . 0))
// indent-tabs-mode: nil
// End:

// vim: filetype=cpp:expandtab:shiftwidth=3:tabstop=3:softtabstop=3