diff options
| -rw-r--r-- | R/R/pca.R | 3 | ||||
| -rw-r--r-- | R/src/wrapper.c | 3 |
2 files changed, 4 insertions, 2 deletions
@@ -1,4 +1,4 @@ -pca <- function(x, k=1, q=0.5, iters=10000){ +pca <- function(x, k=1, q=0.5, iters=10000, nprocs=0){ n <- nrow(x) d <- ncol(x) B <- matrix(0, n, k) @@ -6,6 +6,7 @@ pca <- function(x, k=1, q=0.5, iters=10000){ l <- 0 res <- .C("pca_wrapper", + nprocs=as.integer(nprocs), n=as.integer(n), d=as.integer(d), iters=as.integer(iters), diff --git a/R/src/wrapper.c b/R/src/wrapper.c index 52c1192..5e258b8 100644 --- a/R/src/wrapper.c +++ b/R/src/wrapper.c @@ -1,9 +1,10 @@ #include "pca.h" void -pca_wrapper(int *n, int *d, int *iters, double *q, int *k, double *X, double *B, double *U, double *l) { +pca_wrapper(int *nprocs, int *n, int *d, int *iters, double *q, int *k, double *X, double *B, double *U, double *l) { struct futhark_context_config *cfg = futhark_context_config_new(); + futhark_context_config_set_num_threads(cfg, *nprocs); struct futhark_context *ctx = futhark_context_new(cfg); struct futhark_f64_2d *Xf = futhark_new_f64_2d(ctx, X, *n, *d); |