inst/batchtests/14_stats.R

In pbdDMAT: 'pbdR' Distributed Matrix Methods

suppressPackageStartupMessages(library(pbdDMAT))
init.grid()

comm.print <- function(x) pbdMPI::comm.print(x, quiet=T)

M <- 250
N <- 250
#M <- 6
#N <- 6
BL <- 4
#BL <- 2

comm.set.seed(seed=1234, diff=F)

tol <- 1e-8

# ---------------------------------------------------
# Tests
# ---------------------------------------------------

tests <- function(.)
{
  out1 <- scale(A, T, T)
  out2 <- as.matrix( scale(dA, T, T) )
  comm.print(all.equal(out1, out2))
  
  center <- F
  scale. <- F
  out1 <- prcomp(A, center=center, scale.=scale.)$sdev
  out2 <- prcomp(dA, center=center, scale.=scale.)$sdev
  comm.print(all.equal(out1, out2))
  
  out1 <- cov(A)
  out2 <- as.matrix(cov(dA))
  comm.print(all.equal(out1, out2))
  
  out1 <- cor(A)
  out2 <- as.matrix(cor(dA))
  comm.print(all.equal(out1, out2))
}

# ---------------------------------------------------
# PBLAS
# ---------------------------------------------------

comm.print("-------STATS-------")
comm.print("       Square")

A <- matrix(rnorm(M*N, 10, 100), M, N)
dA <- as.ddmatrix(A, BL)
tests()

comm.print("       Column")

A <- matrix(rnorm(M*1, 10, 100), M, 1)
dA <- as.ddmatrix(A, BL)
tests()

comm.print("       Row")

A <- matrix(rnorm(1*N, 10, 100), 1, N)
dA <- as.ddmatrix(A, BL)
tests()


#comm.print(out1)
#comm.print(out2)

finalize()