Contenitore Cpp NUMA aware

Question

Esiste un modo comodo ed efficiente per utilizzare l'API del contenitore standard cpp in modo sensibile alla NUMA?

Mi piacerebbe fare una moltiplicazione di Matrix Vector sparse parallele OpenMP in un ambiente cpp. per allocare e inizializzare i valori di vettori e matrici per quanto riguarda i domini NUMA, il codice C avrebbe in qualche modo simile a questa:

size_t N = 1000000; 
double* vecVal = malloc (N*sizeof(double)); 

#pragma OMP parallel for 
for (size_i=0; i<N; ++i) 
{ 
    vecVal[i] = 0.; 
} 
/* do spMV */ 
delete vecVal; 

In Cpp Vorrei utilizzare std :: vector (std :: array con anche una dimensione fissa è ok). std :: vector :: reserve() fa il trucco? E 'legale fare qualcosa di simile:

std::vector<double> vec; 
vec.reserve(N); 
double *vecVal = vec.data(); 

#pragma OMP parallel for 
for (size_i=0; i<N; ++i) 
{ 
    vecVal[i] = 0.; 
} 
/* do spMV */ 

Come posso poi impostare il formato corretto per lo std :: vector?

Qualcuno sa un modo più elegante?

Answer 1

Qui è necessario utilizzare uno speciale allocatore numerico. Abbiamo implementato qualcosa di simile per HPX qui: https://github.com/STEllAR-GROUP/hpx/blob/master/hpx/parallel/util/numa_allocator.hpp

L'idea di base è quella di avere il primo tocco all'interno degli allocatori per allocare la funzione. sostituisci i file HPX executor con il tuo ciclo #omp parallel for schedule(static) e dovresti stare bene.