NUMA感知的Cpp容器

Question

有没有一种方便高效的方式，以NUMA感知的方式使用cpp标准容器API？

我想在cpp环境中进行OpenMP并行稀疏矩阵向量乘法。 为了根据NUMA域分配和初始化向量和矩阵值，C代码可能如下所示：

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;

在Cpp中，我想使用std::vector（也可以使用固定大小的std::array）。std::vector::reserve()能够实现这一功能吗？是否可以像这样做：

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 */

我之后该如何为std::vector设置正确的大小？

有没有更优雅的方式呢？

- Titzi

---

resize() 不具备 NUMA 感知能力，因为它会在一个 NUMA 域上初始化所有值（首次接触）。 - Titzi

NUMA感知代码删除malloc分配的内存是否是一件特殊的事情？ - huseyin tugrul buyukisik

Answer 1

您需要在这里使用一个特殊的NUMA感知分配器。我们在这里为HPX实现了类似的东西：https://github.com/STEllAR-GROUP/hpx/blob/master/hpx/parallel/util/numa_allocator.hpp。

基本思想是在分配器的分配函数内部进行第一次接触。用您的#omp parallel for schedule(static)循环替换HPX执行器，那么您就可以顺利完成了。

- Thomas Heller