Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Compute the slave position in the grid on the fly instead of passing it on command...
[simgrid.git] / examples / gras / pmm / pmm.c
index c0320c4..4e398b8 100755 (executable)
@@ -1,4 +1,5 @@
-/* pmm - paralel matrix multiplication "double diffusion"                       */
+/* $Id$ */
+/* pmm - parallel matrix multiplication "double diffusion"                       */
 
 /* Copyright (c) 2006- Ahmed Harbaoui. All rights reserved.                  */
 
@@ -6,8 +7,10 @@
  * under the terms of the license (GNU LGPL) which comes with this package. */
 
 #include "gras.h"
-#define MATRIX_SIZE 3
-#define SLAVE_COUNT 9
+#define PROC_MATRIX_SIZE 3
+#define SLAVE_COUNT (PROC_MATRIX_SIZE*PROC_MATRIX_SIZE)
+
+#define DATA_MATRIX_SIZE 3
 
 XBT_LOG_NEW_DEFAULT_CATEGORY(pmm,"Parallel Matrix Multiplication");
 
@@ -16,14 +19,14 @@ GRAS_DEFINE_TYPE(s_matrix,struct s_matrix {
   int cols;
   double *data GRAS_ANNOTE(size, rows*cols);
 };)
-     typedef struct s_matrix matrix_t;
-
-     /* struct for recovering results */
-     GRAS_DEFINE_TYPE(s_result,struct s_result {
-       int i;
-       int j;
-       double value;
-     });
+typedef struct s_matrix matrix_t;
+
+/* struct for recovering results */
+GRAS_DEFINE_TYPE(s_result,struct s_result {
+  int i;
+  int j;
+  double value;
+});
 typedef struct s_result result_t;
 
 /* struct to send initial data to slave */
@@ -42,7 +45,7 @@ static void register_messages(void) {
   result_type=gras_datadesc_by_symbol(s_result);
   init_data_type=gras_datadesc_by_symbol(s_init_data);
        
-  gras_msgtype_declare("result", result_type);  // recieve a final result from slave
+  gras_msgtype_declare("result", result_type);  // receive 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 
@@ -144,17 +147,17 @@ int master (int argc,char *argv[]) {
   /*  Init the GRAS's infrastructure */
   gras_init(&argc, argv);
 
-  gras_socket_t socket[MATRIX_SIZE*MATRIX_SIZE]; /* sockets for brodcast to other slave */
+  gras_socket_t socket[SLAVE_COUNT]; /* sockets for brodcast to slaves */
 
   /*  Initialize Matrixs */
 
-  A.rows=A.cols=MATRIX_SIZE;
-  B.rows=B.cols=MATRIX_SIZE;
-  C.rows=C.cols=MATRIX_SIZE;
+  A.rows=A.cols=DATA_MATRIX_SIZE;
+  B.rows=B.cols=DATA_MATRIX_SIZE;
+  C.rows=C.cols=DATA_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);
+  A.data=xbt_malloc0(sizeof(double)*DATA_MATRIX_SIZE*DATA_MATRIX_SIZE);
+  B.data=xbt_malloc0(sizeof(double)*DATA_MATRIX_SIZE*DATA_MATRIX_SIZE);
+  C.data=xbt_malloc0(sizeof(double)*DATA_MATRIX_SIZE*DATA_MATRIX_SIZE);
        
   initmatrix(&A);
   initmatrix(&B);
@@ -166,11 +169,13 @@ int master (int argc,char *argv[]) {
   //multiplication();
   //gather();
   //display(A);
-  /****************************** Init Data Send *********************************/
+  /************************* 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){
+  gras_os_sleep(60);      // MODIFIER LES TEMPS D'ATTENTE 60 c trop normalement
+
+  int row=1, col=1;
+  for( i=2;i< argc;i++){
     TRY {
       socket[j]=gras_socket_client(argv[i],port);
     } CATCH(e) {
@@ -178,25 +183,32 @@ int master (int argc,char *argv[]) {
     }
     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.myrow=row;  // My row
+    mydata.mycol=col;  // My column
+    row++;
+    if (row > PROC_MATRIX_SIZE) {
+      row=1;
+      col++;
+    }
                
-    mydata.a=A.data[(mydata.myrow-1)*MATRIX_SIZE+(mydata.mycol-1)];
-    mydata.b=B.data[(mydata.myrow-1)*MATRIX_SIZE+(mydata.mycol-1)];;
+    mydata.a=A.data[(mydata.myrow-1)*PROC_MATRIX_SIZE+(mydata.mycol-1)];
+    mydata.b=B.data[(mydata.myrow-1)*PROC_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);
+    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
+  }
+    // end init Data Send
 
   /******************************* multiplication ********************************/
-  INFO0("begin Multiplication");
+  INFO0("XXXXXXXXXXXXXXXXXXXXXX begin Multiplication");
        
-  for (step=1; step <= MATRIX_SIZE; step++){
-    gras_os_sleep(50);
+  for (step=1; step <= PROC_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*/
+       gras_msg_send(socket[i], gras_msgtype_by_name("step"), &step);
       } CATCH(e) {
        gras_socket_close(socket[i]);
        RETHROW0("Unable to send the msg : %s");
@@ -228,7 +240,7 @@ int master (int argc,char *argv[]) {
   /* 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;
+    C.data[(result.i-1)*DATA_MATRIX_SIZE+(result.j-1)]=result.value;
   }
   /*    end of gather   */
   INFO0 ("The Result of Multiplication is :");
@@ -256,9 +268,11 @@ int slave(int argc,char *argv[]) {
 
   result_t result;
  
-  gras_socket_t from,sock;  /* to recive from server for steps */
+  gras_socket_t from,sock;  /* to receive from server for steps */
 
-  gras_socket_t socket_row[MATRIX_SIZE-1],socket_column[MATRIX_SIZE-1]; /* sockets for brodcast to other slave */
+  /* sockets for brodcast to other slave */
+  gras_socket_t socket_row[PROC_MATRIX_SIZE-1];
+  gras_socket_t socket_column[PROC_MATRIX_SIZE-1];
 
   /* Init the GRAS's infrastructure */
 
@@ -274,9 +288,9 @@ int slave(int argc,char *argv[]) {
   gras_os_sleep(1); //wait to start all slaves 
 
   int i;
-  for (i=1;i<MATRIX_SIZE;i++){
+  for (i=1;i<PROC_MATRIX_SIZE;i++){
     socket_row[i-1]=gras_socket_client(argv[i+1],port);
-    socket_column[i-1]=gras_socket_client(argv[i+MATRIX_SIZE],port);
+    socket_column[i-1]=gras_socket_client(argv[i+PROC_MATRIX_SIZE],port);
   }
 
   /*  Register the known messages */
@@ -294,26 +308,26 @@ int slave(int argc,char *argv[]) {
   mycol=mydata.mycol;
   mydataA=mydata.a;
   mydataB=mydata.b;
-  INFO4("Recive MY POSITION (%d,%d) and MY INIT DATA ( A=%.3g | B=%.3g )",
+  INFO4("Receive 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
+    step=PROC_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);
+    INFO1("Receive a step message from master: step = %d ",step);
 
-    if (step < MATRIX_SIZE ){
+    if (step < PROC_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++){
+       for (l=1;l < PROC_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);  
@@ -327,11 +341,11 @@ int slave(int argc,char *argv[]) {
        } 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));
+       INFO2("Receive 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++){
+       for (l=1;l < PROC_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);
@@ -345,7 +359,7 @@ int slave(int argc,char *argv[]) {
        } 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));
+       INFO2("Receive data A (%.3g) from my neighbor : %s ",bA,gras_socket_peer_name(from));
       }
       bC+=bA*bB;
       INFO1(">>>>>>>> My BC = %.3g",bC);
@@ -356,9 +370,9 @@ int slave(int argc,char *argv[]) {
        
       INFO1("Send ack to master for to end %d th step",step);
     }
-    if(step==MATRIX_SIZE-1) break;
+    if(step==PROC_MATRIX_SIZE-1) break;
        
-  } while (step < MATRIX_SIZE);
+  } while (step < PROC_MATRIX_SIZE);
   /*  wait Message from master to send the result */
  
   result.value=bC;