From f14a1d0b711ab95fe25c326b7c06f3833d2a8d30 Mon Sep 17 00:00:00 2001
From: charleshofer <Charles.Hofer@amd.com>
Date: Wed, 12 Mar 2025 11:30:55 -0500
Subject: [PATCH] Add JSON output to multi-GPU tests (#274)

---
 build/rocm/run_multi_gpu.sh | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/build/rocm/run_multi_gpu.sh b/build/rocm/run_multi_gpu.sh
index 1494f7827..603186377 100755
--- a/build/rocm/run_multi_gpu.sh
+++ b/build/rocm/run_multi_gpu.sh
@@ -54,11 +54,15 @@ run_tests() {
 
     python3 -m pytest \
         --html="${LOG_DIR}/multi_gpu_pmap_test_log.html" \
+        --json-report \
+        --json-report-file="${LOG_DIR}/multi_gpu_pmap_test_log.json" \
         --reruns 3 \
         tests/pmap_test.py
 
     python3 -m pytest \
         --html="${LOG_DIR}/multi_gpu_multi_device_test_log.html" \
+        --json-report \
+        --json-report-file="${LOG_DIR}/multi_gpu_multi_device_test_log.json" \
         --reruns 3 \
         tests/multi_device_test.py