Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
increase matrix size a bit, and do not wait insane amount of time on simulator now...
[simgrid.git] / examples / gras / pmm / pmm.c
index 14c3a4d..7d20b7c 100755 (executable)
-/* 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
+#include "xbt/matrix.h"
+#include "amok/peermanagement.h"
 
-XBT_LOG_NEW_DEFAULT_CATEGORY(pmm,"Messages specific to this example");
+#define PROC_MATRIX_SIZE 3
+#define NEIGHBOR_COUNT PROC_MATRIX_SIZE - 1
+#define SLAVE_COUNT (PROC_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;
+#define DATA_MATRIX_SIZE 18
+const int submatrix_size = DATA_MATRIX_SIZE/PROC_MATRIX_SIZE;
+
+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 sensor */
-GRAS_DEFINE_TYPE(s_init_data,struct s_init_data {
-       int myrow;
-       int mycol;
-       double a;
-       double b;
+/* struct to send initial data to slave */
+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);
-       gras_msgtype_declare("init_data", init_data_type);
+  /* 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);
 
-       gras_msgtype_declare("ask_result", gras_datadesc_by_name("int"));       
-       gras_msgtype_declare("step", gras_datadesc_by_name("int"));
-       gras_msgtype_declare("step_ack", gras_datadesc_by_name("int"));
-       gras_msgtype_declare("data", gras_datadesc_by_name("int"));
+  /* 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 */
-int maestro (int argc,char *argv[]);
-int sensor (int argc,char *argv[]);
+int slave (int argc,char *argv[]);
+int master (int argc,char *argv[]);
+
 
 /* **********************************************************************
- * Maestro code
+ * master code
  * **********************************************************************/
 
 /* Global private data */
 typedef struct {
-  int nbr_col,nbr_row;
+  int nbr_row,nbr_line;
   int remaining_step;
   int remaining_ack;
-} maestro_data_t;
-
-
-static int maestro_cb_data_handler(gras_msg_cb_ctx_t ctx, void *payload) {
+} master_data_t;
 
-  xbt_ex_t e;
-  /* 1. Get the payload into the msg variable */
-  int msg=*(int*)payload_data;
-
-  gras_socket_t expeditor = gras_msg_cb_ctx_from(ctx);
-
-  /*code of callback */
-   
-  /* 8. Make sure we don't leak sockets */
-  gras_socket_close(expeditor);
-   
-  /* 9. Tell GRAS that we consummed this message */
-  return 1;
-} /* end_of_maestro_cb_data_handler */
 
-/***  Function initilaze matrixs ***/
+int master (int argc,char *argv[]) {
 
-static void initmatrix(matrix_t *X){
-       int i;
-       for(i=0 ; i<X.rows*X.cols; i++)
-               X.data[i]=1.0;//1.0*rand()/(RAND_MAX+1.0);
-
-} /* end_of_initmatrixs */
-
-/***  Function Scatter Sequentiel ***/
-
-static void scatter(){
-
-}/* end_of_Scatter */
-
-/***  Function: Scatter // ***/
+  int i;
 
-static void scatter_parl(){
+  xbt_matrix_t A,B,C;
+  result_t result;
 
-}/* end_of_Scatter // */
+  gras_socket_t from;
 
-/***  Function: multiplication ***/
+  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 multiplication(){
-       
-  int step,i;
+  /* 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);
        
-  for (step=1; step <= MATRIX_SIZE; step++){
-    for (i=0; i< nbr_sensor; i++){
-       TRY {
-              gras_msg_send(proc[(i/3)+1][(i%3)+1], gras_msgtype_by_name("step"), &step);  /* initialize Mycol, MyRow, mydataA,mydataB*/
-  
-              myrow,mycol,mydataA,mydataB
-       } CATCH(e) {
-       gras_socket_close(proc[(i/3)+1][(i%3)+1]);
-       RETHROW0("Unable to send the msg : %s");
-       }
-    }
-    /* wait for computing and sensor messages exchange */
-    TRY {
-           gras_msg_wait(600,gras_msgtype_by_name("init_data"),&from,&mydata);
-    } CATCH(e) {
-           RETHROW0("I Can't get a init Data message from Maestro : %s");
-    }
+  /* Create the connexions */
+  xbt_assert0(argc>1, "Usage: master <port>");
+  gras_socket_server(atoi(argv[1]));
+  peers=amok_pm_group_new("pmm");
+   
+  /* friends, we're ready. Come and play */
+  INFO0("Wait for peers for 5 sec");
+  gras_msg_handleall(5);
+  INFO1("Got %ld pals",xbt_dynar_length(peers));
+
+  for (i=0;
+       i<xbt_dynar_length(peers) && i<SLAVE_COUNT;
+       i++) {
+
+    xbt_dynar_get_cpy(peers,i,&grid[i]);
+    INFO2("Connecting to %s:%d.",grid[i]->name,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; i<xbt_dynar_length(peers); ) {
+    xbt_peer_t h;
+
+    xbt_dynar_remove_at(peers,i,&h);
+    INFO2("Too much slaves. Killing %s:%d", h->name, h->port);
+    amok_pm_kill_hp(h->name,h->port);
+    free(h);
   }
 
-}/* end_of_multiplication */
-
-/***  Function: gather ***/
-
-static void gather(){
 
-       
-}/* end_of_gather */
+  /* Assign job to slaves */
+  int row=0, line=0;
+  INFO0("XXXXXXXXXXXXXXXXXXXXXX begin Multiplication");
+  for(i=0 ; i<SLAVE_COUNT; i++){
+    s_pmm_assignment_t assignment;
+    int j,k;
 
-/***  Function: Display Matrix ***/
+    assignment.linepos=line; // assigned line
+    assignment.rowpos=row;   // assigned row
 
-static void display(matrix_t X){
-       
-int i,j,t=0;
-
-  printf("      ");
-  for(j=0;j<X.cols;j++)
-    printf("%.3d ",j);
-    printf("\n");
-    printf("    __");
-    for(j=0;j<X.cols;j++)
-       printf("____");
-       printf("_\n");
-
-       for(i=0;i<X.rows;i++){
-         printf("%.3d | ",i);
-         for(j=0;j<X.cols;j++)
-           printf("%.3g ",X.data[t++]);
-         printf("|\n");
-       }
-       printf("    --");
-       for(j=0;j<X.cols;j++)
-               printf("----");
-       printf("-\n");
-
-}/* end_of_display */
-
-int maestro (int argc,char *argv[]) {
-
-xbt_ex_t e;
-int i,ask_result,step;
-result_t result;
-matrix_t A,B,C;
-gras_socket_t from;
-result_t result;
-
-  gras_socket_t socket[MATRIX_SIZE*MATRIX_SIZE]; /* sockets for brodcast to other sensor */
-       
-
-  /*  Initialize Matrixs */
+    /* Neiborhood */
+    for (j=0,k=0; j<PROC_MATRIX_SIZE; j++) {
+      if (i != j*PROC_MATRIX_SIZE+(row)) {         
+        assignment.row[k] = grid[ j*PROC_MATRIX_SIZE+(row) ] ;
+        k++;
+      }
+    }
+    for (j=0,k=0; j<PROC_MATRIX_SIZE; j++) {
+      if (i != (line)*PROC_MATRIX_SIZE+j) {        
+        assignment.line[k] =  grid[ (line)*PROC_MATRIX_SIZE+j ] ;
+        k++;
+      }
+    }
 
-       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);
-       
-       /*  Init the GRAS's infrastructure */
-       gras_init(&argc, argv);
-       /*  Get arguments and create sockets */
-       port=atoi(argv[1]);
-       //scatter();multiplication();gather();
-       //scatter_parl();
-       /****************************** Init Data Send *********************************/
-       int j=0;
-       init_data_t mydata;
-       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=argv[i+1];  // My rank of row
-               mydata.mycol=argv[i+2];  // My rank of 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)];;
+    assignment.A=xbt_matrix_new_sub(A,
+                                   submatrix_size,submatrix_size,
+                                   submatrix_size*line,submatrix_size*row,
+                                   NULL);
+    assignment.B=xbt_matrix_new_sub(B,
+                                   submatrix_size,submatrix_size,
+                                   submatrix_size*line,submatrix_size*row,
+                                   NULL);
+    row++;
+    if (row >= PROC_MATRIX_SIZE) {
+      row=0;
+      line++;
+    }
                
-               gras_msg_send(socket[j],gras_msgtype_by_name("init_data"),&mydata);
-               j++;
-       } // end init Data Send
+    gras_msg_send(socket[i],gras_msgtype_by_name("pmm_slave"),&assignment);
+    xbt_matrix_free(assignment.A);
+    xbt_matrix_free(assignment.B);
+  }
 
-       /******************************* multiplication ********************************/
+  /* (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   */
 
-       for (step=1; step <= MATRIX_SIZE; step++){
-               for (i=0; i< nbr_sensor; i++){
-               TRY {
-               gras_msg_send(socket[i], gras_msgtype_by_name("step"), &step);  /* initialize Mycol, MyRow, mydataA,mydataB*/
-  
-               myrow,mycol,mydataA,mydataB
-                   } CATCH(e) {
-               gras_socket_close(socket[i]);
-               RETHROW0("Unable to send the msg : %s");
-               }
-       }
-       INFO0("send to sensor to begin a next step");
-       /* wait for computing and sensor messages exchange */
-       i=0;
-       while  ( i< nbr_sensor){
-               TRY {
-               gras_msg_wait(600,gras_msgtype_by_name("step_ack"),&from,&mydata);
-               } CATCH(e) {
-               RETHROW0("I Can't get a init Data message from Maestro : %s");
-               }
-               
-               i++;
-               INFO1("Recive step ack from %s",gras_socket_peer_name(from));
-       }
+  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);
+  }
 
-       }
-       /*********************************  gather ***************************************/
-       
-       int ask_result=0;
-       for( i=1;i< argc;i++){
-               gras_msg_send(socket[i],gras_msgtype_by_name("ask_result"),&ask_result);
-       }
-       /* 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   */
-       display(C);
-
-return 0;
-} /* end_of_maestro */
+  amok_pm_group_shutdown ("pmm");   /* Ok, we're out of here */
+
+  for(i=0; i<SLAVE_COUNT; i++) {
+     gras_socket_close(socket[i]);
+  }
+   
+  xbt_matrix_free(A);
+  xbt_matrix_free(B);
+  xbt_matrix_free(C);
+  gras_exit();
+  return 0;
+} /* end_of_master */
 
 /* **********************************************************************
- * Sensor code
+ * slave code
  * **********************************************************************/
 
-int sensor(int argc,char *argv[]) {
+static int pmm_worker_cb(gras_msg_cb_ctx_t ctx, void *payload) {
+  /* Recover my initialized Data and My Position*/
+  s_pmm_assignment_t assignment = *(s_pmm_assignment_t*)payload;
+  gras_socket_t master = gras_msg_cb_ctx_from(ctx);
 
   xbt_ex_t e; 
 
-  static int bC=0;
-  static int myrow,mycol;
-  static double mydataA,mydataB;
-  int bA,bB;
-  int step,l,result_ack=0;
+  int step,l;
+  xbt_matrix_t bA=xbt_matrix_new(submatrix_size,submatrix_size,
+                                sizeof(double),NULL);
+  xbt_matrix_t bB=xbt_matrix_new(submatrix_size,submatrix_size,
+                                sizeof(double),NULL);
 
-  gras_socket_t from;  /* to recive from server for steps */
+  int myline,myrow;
+  xbt_matrix_t mydataA,mydataB;
+  xbt_matrix_t bC=xbt_matrix_double_new_zeros(submatrix_size,submatrix_size);
+  
+  result_t result;
+  gras_socket_t from;  /* to exchange data with my neighbor */
 
-  gras_socket_t socket_row[2],socket_column[2]; /* sockets for brodcast to other sensor */
+  /* sockets for brodcast to other slave */
+  gras_socket_t socket_line[PROC_MATRIX_SIZE-1];
+  gras_socket_t socket_row[PROC_MATRIX_SIZE-1];
+  memset(socket_line,0,sizeof(socket_line));
+  memset(socket_row,0,sizeof(socket_row));
+   
+  int i;
 
-  /* Init the GRAS's infrastructure */
+  gras_os_sleep(1); /* wait for my pals */
 
-  gras_init(&argc, argv);
+  myline  = assignment.linepos;
+  myrow   = assignment.rowpos;
+  mydataA = assignment.A;
+  mydataB = assignment.B;
 
-  /* Get arguments and create sockets */
+  INFO2("Receive my pos (%d,%d) and assignment",myline,myrow);
 
-  port=atoi(argv[1]);
-  int i;
-  for (i=1;i<MATRIX_SIZE;i++){
-  socket_row[i]=gras_socket_client(argv[i+1],port);
-  socket_column[i]=gras_socket_client(argv[i+MATRIX_SIZE],port);
+  /* Get my neighborhood from the assignment message (skipping myself) */
+  for (i=0 ; i<PROC_MATRIX_SIZE-1 ; i++){
+    socket_line[i]=gras_socket_client(assignment.line[i]->name,
+                                     assignment.line[i]->port);
+    xbt_peer_free(assignment.line[i]);
+  }
+  for (i=0 ; i<PROC_MATRIX_SIZE-1 ; i++){
+    socket_row[i]=gras_socket_client(assignment.row[i]->name,
+                                    assignment.row[i]->port);
+    xbt_peer_free(assignment.row[i]);    
   }
-  INFO2("Launch %s (port=%d)",argv[0],port);
 
-  /*  Create my master socket */
-  sock = gras_socket_server(port);
+  for (step=0; step<PROC_MATRIX_SIZE;step++) {
+       
+    /* a line brodcast */
+    if(myline==step){
+       INFO3("LINE: step(%d) = Myline(%d). Broadcast my data (myport=%d).",
+            step,myline,gras_os_myport());
+       for (l=0;l < PROC_MATRIX_SIZE-1 ;l++) {
+         INFO2("LINE:   Send to %s:%d",
+               gras_socket_peer_name(socket_row[l]),
+               gras_socket_peer_port(socket_row[l]));
+        gras_msg_send(socket_row[l], 
+                      gras_msgtype_by_name("dataB"), 
+                      &mydataB);
+       }
+       
+       
+       xbt_matrix_free(bB);
+       bB = xbt_matrix_new_sub(mydataB,
+                              submatrix_size,submatrix_size,
+                              0,0,NULL);       
+    } else {
+      TRY {
+       xbt_matrix_free(bB);
+       gras_msg_wait(600,gras_msgtype_by_name("dataB"),&from,&bB);
+      } CATCH(e) {
+       RETHROW0("Can't get a data message from line : %s");
+      }
+      INFO4("LINE: step(%d) <> Myline(%d). Receive data from %s:%d",step,myline,
+           gras_socket_peer_name(from), gras_socket_peer_port(from));
+    }
 
-  /*  Register the known messages */
-  register_messages();
+    /* 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);
 
-  /* Recover my initialized Data and My Position*/
-  init_data_t mydata;
+  };
+  /* send Result to master */  
+  result.C=bC;
+  result.linepos=myline;
+  result.rowpos=myrow;
 
   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 Maestro : %s");
+    RETHROW0("Failed to send answer to server: %s");
   }
-  myrow=mydata.myrow;
-  mycol=mydata.mycol;
-  mydataA=mydata.a;
-  mydataB=mydata.b;
+  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]); 
+  }*/
 
-  INFO4("Recover MY POSITION (%d,%d) and MY INIT DATA ( A=%.3g | B=%.3g )",
-       myrow,mycol,mydataA,mydataB);
-
-
-  do {  //repeat until compute Cb
-       step=MATRIX_SIZE+1;  // juste intilization for loop
+  return 1;
+}
 
-  TRY {
-       gras_msg_wait(600,gras_msgtype_by_name("step"),&from,&step);
-  } CATCH(e) {
-         RETHROW0("I Can't get a Next Step message from Maestro : %s");
-  }
+int slave(int argc,char *argv[]) {
+  gras_socket_t mysock;
+  gras_socket_t master;
 
-  /*  Wait for sensors startup */
-  gras_os_sleep(1);
-
-  if (step < MATRIX_SIZE){
-         /* a row brodcast */
-         if(myrow==step){
-                 for (l=1;l < MATRIX_SIZE ;l++){
-                         gras_msg_send(socket_row[l], gras_msgtype_by_name("data"), &mydataB);
-                         bB=mydataB;
-                 }
-         }
-         else
-         {
-                 TRY {
-                         gras_msg_wait(600,gras_msgtype_by_name("data"),
-                                       &from,&bB);
-                 } CATCH(e) {
-                         RETHROW0("I Can't get a data message from row : %s");
-                 }
-         }
-         /* a column brodcast */       
-         if(mycol==step){
-                 for (l=1;l < MATRIX_SIZE ;l++){
-                         gras_msg_send(socket_column[l],gras_msgtype_by_name("data"), &mydataA);
-                         bA=mydataA;
-                 }
-         }
-         else
-         {
-                 TRY {
-                         gras_msg_wait(600,gras_msgtype_by_name("data"),
-                                       &from,&bA);
-                 } CATCH(e) {
-                         RETHROW0("I Can't get a data message from column : %s");
-                 }
-         }
-         bC+=bA*bB;
-         }
-         /* send a ack msg to Maestro */
-       
-         gras_msg_send(from,gras_msgtype_by_name("step_ack"),&step);
-       
-         INFO1("Send ack to maestro for to end %d th step",step);
-       
-         if(step==MATRIX_SIZE-1) break;
+  /* Init the GRAS's infrastructure */
+  gras_init(&argc, argv);
+  amok_pm_init();
 
-  } while (step < MATRIX_SIZE);
-    /*  wait Message from maestro 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");
-         }
-         /* 5. send Result to Maestro */
-         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: %d 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;
-} /* end_of_sensor */
\ No newline at end of file
+  /*  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 */