From d48617672a0388b867a48173b14c618d8834039c Mon Sep 17 00:00:00 2001
From: Josh Romero <joshr@nvidia.com>
Date: Thu, 11 Jun 2026 15:35:47 -0700
Subject: [PATCH] Propogate NVSHMEM compile definitions to ctest build.

Signed-off-by: Josh Romero <joshr@nvidia.com>
---
 tests/ctest/CMakeLists.txt | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/tests/ctest/CMakeLists.txt b/tests/ctest/CMakeLists.txt
index 79b6024..243c56b 100644
--- a/tests/ctest/CMakeLists.txt
+++ b/tests/ctest/CMakeLists.txt
@@ -61,6 +61,16 @@ target_include_directories(cudecomp_test_support
   ${NVHPC_CUDA_INCLUDE_DIR}
   ${NCCL_INCLUDE_DIR}
 )
+if (CUDECOMP_ENABLE_NVSHMEM)
+  target_compile_definitions(cudecomp_test_support
+    PUBLIC
+    ENABLE_NVSHMEM
+  )
+  target_include_directories(cudecomp_test_support
+    PUBLIC
+    ${NVSHMEM_INCLUDE_DIR}
+  )
+endif()
 target_link_libraries(cudecomp_test_support
   PUBLIC
   MPI::MPI_CXX