From ea8c8392ef8a84a146bf1e46fe36d55bdc89799d Mon Sep 17 00:00:00 2001 From: Justin Bedo Date: Fri, 21 Jun 2024 08:53:11 +1000 Subject: R: allow configuration of the number of threads --- R/src/wrapper.c | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'R/src/wrapper.c') 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); -- cgit v1.2.3