From a14803f389c23bd6384df8623eb7e13fc0a8ff9f Mon Sep 17 00:00:00 2001
From: kruskal <99559985+kruskall@users.noreply.github.com>
Date: Fri, 17 Nov 2023 21:29:27 +0100
Subject: [PATCH] test: add cpuprofile arg support to benchmark make task

---
 testing/benchmark/Makefile | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/testing/benchmark/Makefile b/testing/benchmark/Makefile
index 4acf0de03b9..8cf9018b205 100644
--- a/testing/benchmark/Makefile
+++ b/testing/benchmark/Makefile
@@ -79,6 +79,9 @@ apply:
 destroy:
 	@terraform destroy -auto-approve
 
+cp-cpuprof:
+	@[ "${BENCHMARK_CPU_OUT}" ] && scp $(SSH_OPTS) -i $(SSH_KEY) "$(SSH_USER)@$(WORKER_IP):./$(BENCHMARK_CPU_OUT)" $(BENCHMARK_CPU_OUT) || echo "skipping cpu out copy"
+
 .PHONY: log-benckmark-profile
 log-benckmark-profile:
 	@echo "Running benchmarks..."
@@ -93,7 +96,7 @@ log-benckmark-profile:
 run-benchmark: log-benckmark-profile
 	@ssh $(SSH_OPTS) -i $(SSH_KEY) $(SSH_USER)@$(WORKER_IP) ". .envrc && bin/apmbench -run='$(BENCHMARK_RUN)' \
 	-benchtime=$(BENCHMARK_TIME) -count=$(BENCHMARK_COUNT) -warmup-time=$(BENCHMARK_WARMUP_TIME) \
-	-agents=$(BENCHMARK_AGENTS) -detailed=$(BENCHMARK_DETAILED) -event-rate=$(BENCHMARK_EVENT_RATE)" 2>&1 | tee $(BENCHMARK_RESULT)
+	-agents=$(BENCHMARK_AGENTS) -detailed=$(BENCHMARK_DETAILED) -event-rate=$(BENCHMARK_EVENT_RATE) -cpuprofile=$(BENCHMARK_CPU_OUT)" 2>&1 | tee $(BENCHMARK_RESULT)
 
 .PHONY: run-benchmark-autotuned
 run-benchmark-autotuned: