From 0441eec32eb4b12529aeb6c2f6136e3fb7fbc152 Mon Sep 17 00:00:00 2001
From: Wenkai Du <43822138+wenkaidu@users.noreply.github.com>
Date: Mon, 31 Jul 2023 08:11:07 -0700
Subject: [PATCH] p2p_latency_test: clean up IPC temp files at exit (#832)

---
 tools/p2p-latency-test/ll_latency_test.cpp  | 1 +
 tools/p2p-latency-test/p2p_latency_test.cpp | 1 +
 2 files changed, 2 insertions(+)

diff --git a/tools/p2p-latency-test/ll_latency_test.cpp b/tools/p2p-latency-test/ll_latency_test.cpp
index 0f2d5aa987..17a952eb81 100644
--- a/tools/p2p-latency-test/ll_latency_test.cpp
+++ b/tools/p2p-latency-test/ll_latency_test.cpp
@@ -174,6 +174,7 @@ int main(int argc, char** argv) {
       double vega_gpu_rtc_freq = (prop.gcnArch / 10 == 94) ? 1.0E8 : 2.5E7;
       fprintf(stdout, "One-way latency in us: %g\n", double(*time_delta) * 1e6 / NUM_LOOPS_RUN / vega_gpu_rtc_freq / 2);
     }
+    std::remove(file_paths[self_mode]);
   } else {
     fprintf(stderr, "Invalid mode %d\n", self_mode);
     return -1;
diff --git a/tools/p2p-latency-test/p2p_latency_test.cpp b/tools/p2p-latency-test/p2p_latency_test.cpp
index ab16fd3cbf..b9d304f02c 100644
--- a/tools/p2p-latency-test/p2p_latency_test.cpp
+++ b/tools/p2p-latency-test/p2p_latency_test.cpp
@@ -110,6 +110,7 @@ int main(int argc, char** argv) {
       double vega_gpu_rtc_freq = (prop.gcnArch / 10 == 94) ? 1.0E8 : 2.5E7;
       fprintf(stdout, "One-way latency in us: %g\n", double(*time_delta) * 1e6 / NUM_LOOPS_RUN / vega_gpu_rtc_freq / 2);
     }
+    std::remove(file_paths[self_mode]);
   } else {
     fprintf(stderr, "Invalid mode %d\n", self_mode);
     return -1;