X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/5a5c4a0d5915bb7a2b308e061bbf5f5c90e20575..0189615fd0542854c3395490e9c3b635b4b5bba2:/examples/gras/pmm/pmm.c diff --git a/examples/gras/pmm/pmm.c b/examples/gras/pmm/pmm.c index 15412760cd..101ab30c06 100755 --- a/examples/gras/pmm/pmm.c +++ b/examples/gras/pmm/pmm.c @@ -1,55 +1,63 @@ -/* pmm - paralel matrix multiplication "double diffusion" */ +/* $Id$ */ +/* pmm - parallel matrix multiplication "double diffusion" */ -/* Copyright (c) 2006- Ahmed Harbaoui. All rights reserved. */ +/* Copyright (c) 2006 Ahmed Harbaoui. */ +/* Copyright (c) 2006 Martin Quinson. */ +/* All rights reserved. */ /* This program is free software; you can redistribute it and/or modify it * under the terms of the license (GNU LGPL) which comes with this package. */ #include "gras.h" -#define MATRIX_SIZE 3 -#define SLAVE_COUNT 9 +#include "xbt/matrix.h" +#include "amok/peermanagement.h" -XBT_LOG_NEW_DEFAULT_CATEGORY(pmm,"Parallel Matrix Multiplication"); +#define PROC_MATRIX_SIZE 3 +#define NEIGHBOR_COUNT PROC_MATRIX_SIZE - 1 +#define SLAVE_COUNT (PROC_MATRIX_SIZE*PROC_MATRIX_SIZE) + +#define DATA_MATRIX_SIZE 9 +const int submatrix_size = DATA_MATRIX_SIZE/PROC_MATRIX_SIZE; -GRAS_DEFINE_TYPE(s_matrix,struct s_matrix { - int rows; - int cols; - double *data GRAS_ANNOTE(size, rows*cols); -};) -typedef struct s_matrix matrix_t; +XBT_LOG_NEW_DEFAULT_CATEGORY(pmm,"Parallel Matrix Multiplication"); /* struct for recovering results */ GRAS_DEFINE_TYPE(s_result,struct s_result { - int i; - int j; - double value; + int linepos; + int rowpos; + xbt_matrix_t C GRAS_ANNOTE(subtype,double); }); typedef struct s_result result_t; /* struct to send initial data to slave */ -GRAS_DEFINE_TYPE(s_init_data,struct s_init_data { - int myrow; - int mycol; - double a; - double b; +GRAS_DEFINE_TYPE(s_pmm_assignment,struct s_pmm_assignment { + int linepos; + int rowpos; + xbt_peer_t line[NEIGHBOR_COUNT]; + xbt_peer_t row[NEIGHBOR_COUNT]; + xbt_matrix_t A GRAS_ANNOTE(subtype,double); + xbt_matrix_t B GRAS_ANNOTE(subtype,double); }); -typedef struct s_init_data init_data_t; +typedef struct s_pmm_assignment s_pmm_assignment_t; /* register messages which may be sent (common to client and server) */ static void register_messages(void) { - gras_datadesc_type_t result_type; - gras_datadesc_type_t init_data_type; - result_type=gras_datadesc_by_symbol(s_result); - init_data_type=gras_datadesc_by_symbol(s_init_data); + gras_datadesc_type_t result_type; + gras_datadesc_type_t pmm_assignment_type; + + gras_datadesc_set_const("NEIGHBOR_COUNT",NEIGHBOR_COUNT); + result_type=gras_datadesc_by_symbol(s_result); + pmm_assignment_type=gras_datadesc_by_symbol(s_pmm_assignment); - gras_msgtype_declare("result", result_type); // recieve a final result from slave - gras_msgtype_declare("init_data", init_data_type); // send from master to slave to initialize data bA,bB - - gras_msgtype_declare("ask_result", gras_datadesc_by_name("int")); // send from master to slave to ask a final result - gras_msgtype_declare("step", gras_datadesc_by_name("int"));// send from master to slave to indicate the begining of step - gras_msgtype_declare("step_ack", gras_datadesc_by_name("int"));//send from slave to master to indicate the end of the current step - gras_msgtype_declare("dataA", gras_datadesc_by_name("double"));// send data between slave - gras_msgtype_declare("dataB", gras_datadesc_by_name("double"));// send data between slave + /* receive a final result from slave */ + gras_msgtype_declare("result", result_type); + + /* send from master to slave to assign a position and some data */ + gras_msgtype_declare("pmm_slave", pmm_assignment_type); + + /* send data between slaves */ + gras_msgtype_declare("dataA", gras_datadesc_matrix(gras_datadesc_by_name("double"),NULL)); + gras_msgtype_declare("dataB", gras_datadesc_matrix(gras_datadesc_by_name("double"),NULL)); } /* Function prototypes */ @@ -63,327 +71,322 @@ int master (int argc,char *argv[]); /* Global private data */ typedef struct { - int nbr_col,nbr_row; + int nbr_row,nbr_line; int remaining_step; int remaining_ack; } master_data_t; -/*** Function initilaze matrixs ***/ - -static void initmatrix(matrix_t *X){ -int i; - -for(i=0 ; i<(X->rows)*(X->cols); i++) - X->data[i]=1.0;//*rand()/(RAND_MAX+1.0); -} /* end_of_initmatrixs */ - -/*** Function Scatter Sequentiel ***/ - -static void scatter(){ - -}/* end_of_Scatter */ - -/*** Function: Scatter // ***/ - -static void scatter_parl(){ - -}/* end_of_Scatter // */ - -/*** Function: multiplication ***/ - -static void multiplication(){ - -}/* end_of_multiplication */ +int master (int argc,char *argv[]) { -/*** Function: gather ***/ + int i; + double init_delay; -static void gather(){ + xbt_matrix_t A,B,C; + result_t result; -}/* end_of_gather */ + gras_socket_t from; -/*** Function: Display Matrix ***/ + xbt_dynar_t peers; /* group of slaves */ + xbt_peer_t grid[SLAVE_COUNT]; /* The slaves as an array */ + gras_socket_t socket[SLAVE_COUNT]; /* sockets for brodcast to slaves */ -static void display(matrix_t X){ + /* Init the GRAS's infrastructure */ + gras_init(&argc, argv); + amok_pm_init(); + register_messages(); + + /* Initialize data matrices */ + A = xbt_matrix_double_new_id(DATA_MATRIX_SIZE,DATA_MATRIX_SIZE); + B = xbt_matrix_double_new_seq(DATA_MATRIX_SIZE,DATA_MATRIX_SIZE); + C = xbt_matrix_double_new_zeros(DATA_MATRIX_SIZE,DATA_MATRIX_SIZE); -int i,j,t=0; - - printf(" "); - for(j=0;j1, "Usage: master "); + gras_socket_server(atoi(argv[1])); + peers=amok_pm_group_new("pmm"); + + /* friends, we're ready. Come and play */ + if (gras_if_RL()) { + init_delay = 5; + } else { + init_delay = 15; /* no idea why I have to wait that long in simulation */ + } + INFO1("Wait for peers for %.0f sec",init_delay); + gras_msg_handleall(init_delay); + INFO1("Got %ld pals",xbt_dynar_length(peers)); + + for (i=0; + iname,grid[i]->port); + socket[i]=gras_socket_client(grid[i]->name,grid[i]->port); + INFO2("Connected to %s:%d.",grid[i]->name,grid[i]->port); + } + xbt_assert2(i==SLAVE_COUNT, + "Not enough slaves for this setting (got %d of %d). Change the deployment file", + i,SLAVE_COUNT); + + /* Kill surnumerous slaves */ + for (i=SLAVE_COUNT; iname, h->port); + amok_pm_kill_hp(h->name,h->port); + free(h); + } -gras_socket_t from; - /* Init the GRAS's infrastructure */ - gras_init(&argc, argv); + /* Assign job to slaves */ + int row=0, line=0; + INFO0("XXXXXXXXXXXXXXXXXXXXXX begin Multiplication"); + for(i=0 ; i= PROC_MATRIX_SIZE) { + row=0; + line++; + } + + gras_msg_send(socket[i],gras_msgtype_by_name("pmm_slave"),&assignment); + xbt_matrix_free(assignment.A); + xbt_matrix_free(assignment.B); + } - gras_socket_t socket[MATRIX_SIZE*MATRIX_SIZE]; /* sockets for brodcast to other slave */ + /* (have a rest while the slave perform the multiplication) */ + + /* Retrieve the results */ + for( i=0;i< SLAVE_COUNT;i++){ + gras_msg_wait(6000,gras_msgtype_by_name("result"),&from,&result); + VERB2("%d slaves are done already. Waiting for %d",i+1, SLAVE_COUNT); + xbt_matrix_copy_values(C,result.C, submatrix_size,submatrix_size, + submatrix_size*result.linepos, + submatrix_size*result.rowpos, + 0,0,NULL); + xbt_matrix_free(result.C); + } + /* end of gather */ - /* Initialize Matrixs */ + if (DATA_MATRIX_SIZE < 30) { + INFO0 ("The Result of Multiplication is :"); + xbt_matrix_dump(C,"C:res",0,xbt_matrix_dump_display_double); + } else { + INFO1("Matrix size too big (%d>30) to be displayed here",DATA_MATRIX_SIZE); + } - A.rows=A.cols=MATRIX_SIZE; - B.rows=B.cols=MATRIX_SIZE; - C.rows=C.cols=MATRIX_SIZE; - - A.data=xbt_malloc0(sizeof(double)*MATRIX_SIZE*MATRIX_SIZE); - B.data=xbt_malloc0(sizeof(double)*MATRIX_SIZE*MATRIX_SIZE); - C.data=xbt_malloc0(sizeof(double)*MATRIX_SIZE*MATRIX_SIZE); - - initmatrix(&A); - initmatrix(&B); - - /* Get arguments and create sockets */ - port=atoi(argv[1]); - //scatter(); - //scatter_parl(); - //multiplication(); - //gather(); - //display(A); - /****************************** Init Data Send *********************************/ - int step_ack,j=0; - init_data_t mydata; - gras_os_sleep(60); // MODIFIER LES TEMPS D'ATTENTE 60 c trop normalement - for( i=2;i< argc;i+=3){ - TRY { - socket[j]=gras_socket_client(argv[i],port); - } CATCH(e) { - RETHROW0("Unable to connect to the server: %s"); - } - INFO2("Connected to %s:%d.",argv[i],port); - - mydata.myrow=atoi(argv[i+1]); // My row - mydata.mycol=atoi(argv[i+2]); // My column - - mydata.a=A.data[(mydata.myrow-1)*MATRIX_SIZE+(mydata.mycol-1)]; - mydata.b=B.data[(mydata.myrow-1)*MATRIX_SIZE+(mydata.mycol-1)];; - - gras_msg_send(socket[j],gras_msgtype_by_name("init_data"),&mydata); - INFO3("Send Init Data to %s : data A= %.3g & data B= %.3g",gras_socket_peer_name(socket[j]),mydata.a,mydata.b); - j++; - } // end init Data Send + amok_pm_group_shutdown ("pmm"); /* Ok, we're out of here */ - /******************************* multiplication ********************************/ - INFO0("begin Multiplication"); - - for (step=1; step <= MATRIX_SIZE; step++){ - gras_os_sleep(50); - for (i=0; i< SLAVE_COUNT; i++){ - TRY { - gras_msg_send(socket[i], gras_msgtype_by_name("step"), &step); /* initialize Mycol, MyRow, mydataA,mydataB*/ - } CATCH(e) { - gras_socket_close(socket[i]); - RETHROW0("Unable to send the msg : %s"); - } - } - INFO1("send to slave to begin a %d th step",step); - /* wait for computing and slave messages exchange */ - i=0; - - while ( i< SLAVE_COUNT){ - TRY { - gras_msg_wait(1300,gras_msgtype_by_name("step_ack"),&from,&step_ack); - } CATCH(e) { - RETHROW0("I Can't get a Ack step message from slave : %s"); - } - i++; - INFO3("Receive Ack step ack from %s (got %d of %d)", - gras_socket_peer_name(from), - i, SLAVE_COUNT); - } - } - /********************************* gather ***************************************/ - - ask_result=0; - for( i=1;i< argc;i++){ - gras_msg_send(socket[i],gras_msgtype_by_name("ask_result"),&ask_result); - INFO1("Send (Ask Result) message to %s",gras_socket_peer_name(socket[i])); - } - /* wait for results */ - for( i=1;i< argc;i++){ - gras_msg_wait(600,gras_msgtype_by_name("result"),&from,&result); - C.data[(result.i-1)*MATRIX_SIZE+(result.j-1)]=result.value; - } - /* end of gather */ - INFO0 ("The Result of Multiplication is :"); - display(C); - -return 0; + for(i=0; iname, + assignment.line[i]->port); + xbt_peer_free(assignment.line[i]); + } + for (i=0 ; iname, + assignment.row[i]->port); + xbt_peer_free(assignment.row[i]); } - /* Register the known messages */ - register_messages(); + for (step=0; step Myline(%d). Receive data from %s:%d",step,myline, + gras_socket_peer_name(from), gras_socket_peer_port(from)); + } + + /* a row brodcast */ + if (myrow==step) { + INFO2("ROW: step(%d)=myrow(%d). Broadcast my data",step,myrow); + for (l=1;l < PROC_MATRIX_SIZE ; l++) { + INFO2("ROW: Send to %s:%d", + gras_socket_peer_name(socket_line[l-1]), + gras_socket_peer_port(socket_line[l-1])); + gras_msg_send(socket_line[l-1],gras_msgtype_by_name("dataA"), &mydataA); + } + xbt_matrix_free(bA); + bA = xbt_matrix_new_sub(mydataA, + submatrix_size,submatrix_size, + 0,0,NULL); + } else { + TRY { + xbt_matrix_free(bA); + gras_msg_wait(1200,gras_msgtype_by_name("dataA"), &from,&bA); + } CATCH(e) { + RETHROW0("Can't get a data message from row : %s"); + } + INFO3("ROW: step(%d)<>myrow(%d). Receive data from %s",step,myrow, + gras_socket_peer_name(from)); + } + xbt_matrix_double_addmult(bA,bB,bC); + + }; + + /* send Result to master */ + result.C=bC; + result.linepos=myline; + result.rowpos=myrow; - /* Recover my initialized Data and My Position*/ - init_data_t mydata; - INFO0("wait for init Data"); TRY { - gras_msg_wait(600,gras_msgtype_by_name("init_data"),&from,&mydata); + gras_msg_send(master, gras_msgtype_by_name("result"),&result); } CATCH(e) { - RETHROW0("I Can't get a init Data message from master : %s"); + RETHROW0("Failed to send answer to server: %s"); } - myrow=mydata.myrow; - mycol=mydata.mycol; - mydataA=mydata.a; - mydataB=mydata.b; - INFO4("Recive MY POSITION (%d,%d) and MY INIT DATA ( A=%.3g | B=%.3g )", - myrow,mycol,mydataA,mydataB); - step=1; - - do { //repeat until compute Cb - step=MATRIX_SIZE+1; // just intilization for loop - - TRY { - gras_msg_wait(200,gras_msgtype_by_name("step"),&from,&step); - } CATCH(e) { - RETHROW0("I Can't get a Next Step message from master : %s"); - } - INFO1("Recive a step message from master: step = %d ",step); - - if (step < MATRIX_SIZE ){ - /* a row brodcast */ - gras_os_sleep(3); // IL FAUT EXPRIMER LE TEMPS D'ATTENTE EN FONCTION DE "SLAVE_COUNT" - if(myrow==step){ - INFO2("step(%d) = Myrow(%d)",step,myrow); - for (l=1;l < MATRIX_SIZE ;l++){ - gras_msg_send(socket_column[l-1], gras_msgtype_by_name("dataB"), &mydataB); - bB=mydataB; - INFO1("send my data B (%.3g) to my (vertical) neighbors",bB); - } - } - if(myrow != step){ - INFO2("step(%d) <> Myrow(%d)",step,myrow); - TRY { - gras_msg_wait(600,gras_msgtype_by_name("dataB"), - &from,&bB); - } CATCH(e) { - RETHROW0("I Can't get a data message from row : %s"); - } - INFO2("Recive data B (%.3g) from my neighbor: %s",bB,gras_socket_peer_name(from)); - } - /* a column brodcast */ - if(mycol==step){ - for (l=1;l < MATRIX_SIZE ;l++){ - gras_msg_send(socket_row[l-1],gras_msgtype_by_name("dataA"), &mydataA); - bA=mydataA; - INFO1("send my data A (%.3g) to my (horizontal) neighbors",bA); - } - } - - if(mycol != step){ - TRY { - gras_msg_wait(1200,gras_msgtype_by_name("dataA"), - &from,&bA); - } CATCH(e) { - RETHROW0("I Can't get a data message from column : %s"); - } - INFO2("Recive data A (%.3g) from my neighbor : %s ",bA,gras_socket_peer_name(from)); - } - bC+=bA*bB; - INFO1(">>>>>>>> My BC = %.3g",bC); - - /* send a ack msg to master */ - - gras_msg_send(from,gras_msgtype_by_name("step_ack"),&step); - - INFO1("Send ack to master for to end %d th step",step); - } - if(step==MATRIX_SIZE-1) break; - - } while (step < MATRIX_SIZE); - /* wait Message from master to send the result */ - - result.value=bC; - result.i=myrow; - result.j=mycol; - - TRY { - gras_msg_wait(600,gras_msgtype_by_name("ask_result"), - &from,&result_ack); - } CATCH(e) { - RETHROW0("I Can't get a data message from row : %s"); - } - /* send Result to master */ - TRY { - gras_msg_send(from, gras_msgtype_by_name("result"),&result); - } CATCH(e) { - // gras_socket_close(from); - RETHROW0("Failed to send PING to server: %s"); - } - INFO3(">>>>>>>> Result: %.3f sent to %s:%d <<<<<<<<", - bC, - gras_socket_peer_name(from),gras_socket_peer_port(from)); - /* Free the allocated resources, and shut GRAS down */ - gras_socket_close(from); - gras_exit(); - INFO0("Done."); - return 0; + INFO2(">>>>>>>> Result sent to %s:%d <<<<<<<<", + gras_socket_peer_name(master),gras_socket_peer_port(master)); + /* Free the allocated resources, and shut GRAS down */ + + xbt_matrix_free(bA); + xbt_matrix_free(bB); + xbt_matrix_free(bC); + + xbt_matrix_free(mydataA); + xbt_matrix_free(mydataB); + gras_socket_close(master); + gras_socket_close(from); + /* FIXME: some are said to be unknown + for (l=0; l < PROC_MATRIX_SIZE-1; l++) { + if (socket_line[l]) + gras_socket_close(socket_line[l]); + if (socket_row[l]) + gras_socket_close(socket_row[l]); + }*/ + + return 1; +} + +int slave(int argc,char *argv[]) { + gras_socket_t mysock; + gras_socket_t master; + + /* Init the GRAS's infrastructure */ + gras_init(&argc, argv); + amok_pm_init(); + + /* Register the known messages and my callback */ + register_messages(); + gras_cb_register(gras_msgtype_by_name("pmm_slave"),pmm_worker_cb); + + /* Create the connexions */ + mysock = gras_socket_server_range(3000,9999,0,0); + gras_os_sleep(1); /* let the master get ready */ + INFO1("Sensor starting (on port %d)",gras_os_myport()); + master = gras_socket_client_from_string(argv[1]); + + /* Join and run the group */ + amok_pm_group_join(master,"pmm"); + amok_pm_mainloop(600); + + /* housekeeping */ + gras_socket_close(mysock); + // gras_socket_close(master); Unknown + gras_exit(); + return 0; } /* end_of_slave */