rdma: make the mempool size configurable (#8)

bytedance · Oct 28, 2019 · 35e3d13 · 35e3d13
1 parent 862f8a6
commit 35e3d13
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/rdma_van.h b/src/rdma_van.h
@@ -68,6 +68,12 @@ class SimpleMempool {
     std::lock_guard<std::mutex> lk(mu_);
     pd_ = pd;
     struct ibv_mr *mr;
+
+    // set init mempool size
+    auto val = Environment::Get()->find("BYTEPS_RDMA_MEMPOOL_SIZE");
+    size = val ? atoi(val) : size;
+    PS_VLOG(1) << "RDMA initial mempool size set to " << size;
+
     char *p = reinterpret_cast<char *>(aligned_alloc(kAlignment, size));
     total_allocated_size += size;
     CHECK(p);