load balance

Author: snoweye

DESCRIPTION

@@ -25,7 +25,7 @@ URL: http://r-pbd.org/
 BugReports: http://group.r-pbd.org/
 MailingList: Please send questions and comments regarding pbdR to
         [email protected]
-Packaged: 2013-07-06 20:13:52 UTC; snoweye
+Packaged: 2013-07-09 14:46:57 UTC; snoweye
 Author: Wei-Chen Chen [aut, cre],
   George Ostrouchov [aut]
 Maintainer: Wei-Chen Chen <[email protected]>

R/02_load_balance.r

@@ -82,20 +82,22 @@ load.balance <- function(X.spmd, bal.info = NULL, comm = .SPMD.CT$comm,
     stop("spmd.major = 1 or 2.")
   }
 
+  storage.mode(X.spmd) <- "double"
+
   send.to <- as.integer(unique(bal.info$send$belong))
   if(length(send.to) > 0){
     if(spmd.major == 1){
       for(i in send.to){
         if(i != COMM.RANK){
           tmp <- matrix(X.spmd[bal.info$send$belong == i,], ncol = p)
-          send(tmp, rank.dest = i, tag = COMM.RANK, comm = comm)
+          spmd.isend.double(tmp, rank.dest = i, tag = COMM.RANK, comm = comm)
         }
       }
     } else{
       for(i in send.to){
         if(i != COMM.RANK){
           tmp <- matrix(X.spmd[, bal.info$send$belong == i], nrow = p)
-          send(tmp, rank.dest = i, tag = COMM.RANK, comm = comm)
+          spmd.isend.double(tmp, rank.dest = i, tag = COMM.RANK, comm = comm)
         }
       }
     }
@@ -107,8 +109,10 @@ load.balance <- function(X.spmd, bal.info = NULL, comm = .SPMD.CT$comm,
     if(spmd.major == 1){
       for(i in recv.from){
         if(i != COMM.RANK){
-          tmp <- recv(rank.source = i, tag = i, comm = comm)
-          dim(tmp) <- c(length(tmp) / p, p)
+          total.row <- sum(bal.info$recv$org == i)
+          tmp <- spmd.recv.double(double(total.row * p),
+                                  rank.source = i, tag = i, comm = comm)
+          dim(tmp) <- c(total.row, p)
         } else{
           tmp <- matrix(X.spmd[bal.info$send$belong == i,], ncol = p)
         }
@@ -117,8 +121,10 @@ load.balance <- function(X.spmd, bal.info = NULL, comm = .SPMD.CT$comm,
     } else{
       for(i in recv.from){
         if(i != COMM.RANK){
-          tmp <- recv(rank.source = i, tag = i, comm = comm)
-          dim(tmp) <- c(p, length(tmp) / p)
+          total.column <- sum(bal.info$recv$org == i)
+          tmp <- spmd.recv.double(double(total.column * p),
+                                  rank.source = i, tag = i, comm = comm)
+          dim(tmp) <- c(p, total.column)
         } else{
           tmp <- matrix(X.spmd[, bal.info$send$belong == i], nrow = p)
         }
@@ -137,6 +143,8 @@ load.balance <- function(X.spmd, bal.info = NULL, comm = .SPMD.CT$comm,
     }
   }
 
+  spmd.wait()
+
   ret
 } # End of load.balance().