From b08b2c4c2a51e9e49c6d38a153ca5e0269e8272b Mon Sep 17 00:00:00 2001 From: Birte Kristina Friesel Date: Thu, 4 Jul 2024 09:40:07 +0200 Subject: GEMV: Add NUMA variant for HBM evaluation --- GEMV/baselines/cpu/gemv_utils.h | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'GEMV/baselines/cpu/gemv_utils.h') diff --git a/GEMV/baselines/cpu/gemv_utils.h b/GEMV/baselines/cpu/gemv_utils.h index 605f148..c99763a 100644 --- a/GEMV/baselines/cpu/gemv_utils.h +++ b/GEMV/baselines/cpu/gemv_utils.h @@ -1,7 +1,16 @@ void allocate_dense(size_t rows,size_t cols, double*** dense) { +#if NUMA + if (bitmask_in) { + numa_set_membind(bitmask_in); + numa_free_nodemask(bitmask_in); + } + *dense = numa_alloc(sizeof(double)*rows); + **dense = numa_alloc(sizeof(double)*rows*cols); +#else *dense = malloc(sizeof(double)*rows); **dense = malloc(sizeof(double)*rows*cols); +#endif for (size_t i=0; i < rows; i++ ) { (*dense)[i] = (*dense)[0] + i*cols; -- cgit v1.2.3