Not reserving mempool

mpc-msri · May 29, 2024 · 8d7f83c · 8d7f83c
1 parent 78cd153
commit 8d7f83c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/GPU-MPC/backend/sigma.h b/GPU-MPC/backend/sigma.h
@@ -66,10 +66,11 @@ class SIGMA : public Backend<T>
     std::vector<GroupElement> *invSqrtTab;
     LlamaTransformer<T> *llama;
 
-    SIGMA(int party, std::string ip, std::string keyFile, int bw, int scale, int n_seq, int n_embed, int numThreads) : party(party), bw(bw), scale(scale), n_seq(n_seq)
+    SIGMA(int party, std::string ip, std::string keyFile, int bw, int scale, int n_seq, int n_embed, int numThreads, bool initGPUMemPool = true) : party(party), bw(bw), scale(scale), n_seq(n_seq)
     {
         initAESContext(&g);
-        initGPUMemPool();
+        if (initGPUMemPool)
+            initGPUMemPool();
         // initCommBufs(true);
 
         d_geluTab = genLUT<T, reluSubGelu<T>>(8, 6, scale);

diff --git a/GPU-MPC/experiments/sigma/sigma.cu b/GPU-MPC/experiments/sigma/sigma.cu
@@ -193,7 +193,7 @@ int main(int __argc, char **__argv)
     statsFile.close();
 
     std::string ip(__argv[4]);
-    auto sigma = new SIGMA<u64>(party, ip, "", bw, scale, n_seq, n_embd, atoi(__argv[5]));
+    auto sigma = new SIGMA<u64>(party, ip, "", bw, scale, n_seq, n_embd, atoi(__argv[5]), false);
     sigma->keyBuf = sigmaKeygen->startPtr;
     sigma->startPtr = sigma->keyBuf;
     sigma->keySize = sigmaKeygen->keySize;