add 8GB memory limit for streaming

harsha-simhadri · Oct 15, 2023 · 1c3268e · 1c3268e
1 parent d6ffcd8
commit 1c3268e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmark/runner.py b/benchmark/runner.py
@@ -265,7 +265,7 @@ def run_docker(definition, dataset, count, runs, timeout, rebuild,
 
     client = docker.from_env()
     if mem_limit is None:
-        mem_limit = psutil.virtual_memory().available
+        mem_limit = psutil.virtual_memory().available if neurips23track != 'streaming' else ll(8*1024*1024*1024)
 
     # ready the container object invoked later in this function
     container = None