Nothing
### Median level functions for R objects. These should not be in S3/S4.
### For general types.
spmd.gather.object <- function(x,
rank.dest = .pbd_env$SPMD.CT$rank.root, comm = .pbd_env$SPMD.CT$comm,
unlist = .pbd_env$SPMD.CT$unlist){
x.raw <- serialize(x, NULL)
x.count <- spmd.allgather.integer(length(x.raw),
integer(spmd.comm.size(comm)),
comm = comm)
displs <- c(0L, cumsum(x.count))
### mpi_allgatherv() in C requires displs[1:(length(displs) - 1)] only.
### It only passes the pointer, so displs[1:length(displs)] is OK, too.
ret <- spmd.gatherv.raw(x.raw, raw(sum(x.count)), x.count = x.count,
displs = displs,
rank.dest = rank.dest, comm = comm)
if(spmd.comm.rank(comm) != rank.dest){
return(invisible())
}
ret <- lapply(1:length(x.count),
function(i) unserialize(ret[(displs[i] + 1):displs[i + 1]]))
if(unlist){
ret <- unlist(ret)
}
ret
} # End of spmd.gather.object().
### For array only.
spmd.gather.array <- function(x,
rank.dest = .pbd_env$SPMD.CT$rank.root, comm = .pbd_env$SPMD.CT$comm,
unlist = .pbd_env$SPMD.CT$unlist){
n.dim <- length(dim(x))
COMM.SIZE <- spmd.comm.size(comm)
all.check <- spmd.allreduce.integer(
as.integer(is.double(x) && length(x) > 0),
integer(1), op = "sum", comm = comm) == COMM.SIZE
if(all.check){
x.dim <- spmd.allgather.integer(dim(x), integer(COMM.SIZE * n.dim),
comm = comm)
dim(x.dim) <- c(n.dim, COMM.SIZE)
x.count <- as.integer(apply(x.dim, 2, prod))
displs <- c(0L, cumsum(x.count))
ret <- spmd.gatherv.double(x, double(sum(x.count)), x.count,
displs = displs,
rank.dest = rank.dest, comm = comm)
if(spmd.comm.rank(comm) != rank.dest){
return(invisible())
}
ret <- lapply(1:COMM.SIZE,
function(i){
array(data = ret[(displs[i] + 1):displs[i + 1]],
dim = x.dim[, i])
})
if(unlist){
ret <- unlist(ret)
}
return(ret)
}
all.check <- spmd.allreduce.integer(
as.integer(is.integer(x) && length(x) > 0),
integer(1), op = "sum", comm = comm) == COMM.SIZE
if(all.check){
x.dim <- spmd.allgather.integer(dim(x), integer(COMM.SIZE * n.dim),
comm = comm)
dim(x.dim) <- c(n.dim, COMM.SIZE)
x.count <- as.integer(apply(x.dim, 2, prod))
displs <- c(0L, cumsum(x.count))
ret <- spmd.gatherv.integer(x, integer(sum(x.count)), x.count,
displs = displs,
rank.dest = rank.dest, comm = comm)
if(spmd.comm.rank(comm) != rank.dest){
return(invisible())
}
ret <- lapply(1:COMM.SIZE,
function(i){
array(data = ret[(displs[i] + 1):displs[i + 1]],
dim = x.dim[, i])
})
if(unlist){
ret <- unlist(ret)
}
return(ret)
}
spmd.gather.object(x, rank.dest = rank.dest, comm = comm,
unlist = unlist)
} # End of spmd.gather.array().
Any scripts or data that you put into this service are public.
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.