From 5ed5c4642ef9b9bc142a932d54ccb0ab9226d07a Mon Sep 17 00:00:00 2001
From: Yiltan Temucin <yiltan@amd.com>
Date: Thu, 14 Nov 2024 15:34:33 -0600
Subject: [PATCH] Explicitly require rocPRIM and rocThrust.

[ROCm/rocshmem commit: 50e46847c60eca143053328c2cc19a01bf9136f9]
---
 projects/rocshmem/tests/unit_tests/CMakeLists.txt | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/projects/rocshmem/tests/unit_tests/CMakeLists.txt b/projects/rocshmem/tests/unit_tests/CMakeLists.txt
index b47da1c154..568b156fd2 100644
--- a/projects/rocshmem/tests/unit_tests/CMakeLists.txt
+++ b/projects/rocshmem/tests/unit_tests/CMakeLists.txt
@@ -115,6 +115,10 @@ target_link_libraries(
 ###############################################################################
 # GTEST DEPENDENCY
 ###############################################################################
+# These packages are required for the unit tests
+find_package(rocprim REQUIRED CONFIG PATHS "${ROCM_PATH}/include/rocprim")
+find_package(rocthrust REQUIRED CONFIG PATHS "${ROCM_PATH}/include/rocthrust")
+
 include(FetchContent)
 
 FetchContent_Declare(
@@ -135,4 +139,5 @@ target_link_libraries(
   PRIVATE
     gtest
     gtest_main
+    roc::rocthrust
 )