From 3cf8f17f3f4c6df8868f12c07a15409824315ff3 Mon Sep 17 00:00:00 2001
From: Arthur Hanson Rasmusson <arthur@vgpu.io>
Date: Wed, 30 Sep 2026 08:40:17 -0400
Subject: guest-diag: cdt2 mapping scan is opt-in (CDT_SCAN=1); it can wedge
 the client

---
 tools/test/cuda_drv_test2.c | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

(limited to 'tools')

diff --git a/tools/test/cuda_drv_test2.c b/tools/test/cuda_drv_test2.c
index 98a996e..a3d3190 100644
--- a/tools/test/cuda_drv_test2.c
+++ b/tools/test/cuda_drv_test2.c
@@ -175,11 +175,13 @@ int main(void)
     MARK("cuMemcpyHtoD");
     CHECK(cuMemcpyHtoD(dptr, buf, sizeof(buf)));
     pause_env("CDT_HP_PAUSE");
-    scan_self_for_payload("after-HtoD");
+    if (getenv("CDT_SCAN"))
+        scan_self_for_payload("after-HtoD");
     memset(back, 0, sizeof(back));
     MARK("cuMemcpyDtoH");
     CHECK(cuMemcpyDtoH(back, dptr, sizeof(buf)));
-    scan_self_for_payload("after-DtoH");
+    if (getenv("CDT_SCAN"))
+        scan_self_for_payload("after-DtoH");
     {
         size_t matched = 0;
         for (i = 0; i < 64; i++)
-- 
cgit v1.2.3