add 8GB memory limit for streaming

harsha-simhadri · Oct 15, 2023 · 6a615b8 · 6a615b8
1 parent fe8af53
commit 6a615b8
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 (8*1024*1024*1024)
 
     # ready the container object invoked later in this function
     container = None