Install the things required to collect flamegraphs when needed

harsha-simhadri · Sep 10, 2024 · 3f65080 · 3f65080
1 parent 9b68a2a
commit 3f65080
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/benchmark/runner.py b/benchmark/runner.py
@@ -289,6 +289,7 @@ def run_docker(definition, dataset, count, runs, timeout, rebuild,
             },
             cpuset_cpus=cpu_limit,
             mem_limit=mem_limit,
+            privileged=True, # so that we can run perf inside the container
             detach=True)
 
     # set/override container timeout based on competition flag

diff --git a/neurips23/streaming/Dockerfile.BasePostgres b/neurips23/streaming/Dockerfile.BasePostgres
@@ -19,3 +19,9 @@ USER root
 # copy the script to start the database that we will use start the database during the tests
 COPY neurips23/streaming/start_database.sh /home/app/
 RUN chmod +x /home/app/start_database.sh
+
+# install linux-tools-generic into docker so that devs can use perf if they want
+RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt install -y linux-tools-generic
+
+# clone FlameGraph for the same purpose
+RUN git clone https://github.com/brendangregg/FlameGraph