From 86d24a8bed9983452b09ede68242ee7682e05a52 Mon Sep 17 00:00:00 2001
From: Aryan Salmanpour <aryan.salmanpour@amd.com>
Date: Mon, 13 Jan 2025 14:51:54 -0500
Subject: [PATCH] Update samples (#108)

---
 samples/jpegDecode/CMakeLists.txt        | 8 +++++++-
 samples/jpegDecodeBatched/CMakeLists.txt | 8 +++++++-
 samples/jpegDecodePerf/CMakeLists.txt    | 8 +++++++-
 3 files changed, 21 insertions(+), 3 deletions(-)

diff --git a/samples/jpegDecode/CMakeLists.txt b/samples/jpegDecode/CMakeLists.txt
index 3695487afc..4e129c5557 100644
--- a/samples/jpegDecode/CMakeLists.txt
+++ b/samples/jpegDecode/CMakeLists.txt
@@ -46,8 +46,9 @@ list(APPEND CMAKE_PREFIX_PATH ${ROCM_PATH}/hip ${ROCM_PATH})
 
 find_package(HIP QUIET)
 find_package(rocJPEG QUIET)
+find_package(rocprofiler-register QUIET)
 
-if(HIP_FOUND AND ROCJPEG_FOUND)
+if(HIP_FOUND AND ROCJPEG_FOUND AND rocprofiler-register_FOUND)
     # HIP
     set(LINK_LIBRARY_LIST ${LINK_LIBRARY_LIST} hip::host)
     # rocJPEG
@@ -55,6 +56,8 @@ if(HIP_FOUND AND ROCJPEG_FOUND)
     set(LINK_LIBRARY_LIST ${LINK_LIBRARY_LIST} ${ROCJPEG_LIBRARY})
     # std filesystem
     set(LINK_LIBRARY_LIST ${LINK_LIBRARY_LIST} stdc++fs)
+    # rocprofiler-register
+    set(LINK_LIBRARY_LIST ${LINK_LIBRARY_LIST} rocprofiler-register::rocprofiler-register)
 
     list(APPEND SOURCES ${PROJECT_SOURCE_DIR} jpegdecode.cpp)
     add_executable(${PROJECT_NAME} ${SOURCES})
@@ -68,4 +71,7 @@ else()
     if (NOT ROCJPEG_FOUND)
         message(FATAL_ERROR "-- ERROR!: rocJPEG Not Found! - please install rocJPEG!")
     endif()
+    if (NOT rocprofiler-register_FOUND)
+        message(FATAL_ERROR "-- ERROR!: rocprofiler-register Not Found! - please install rocprofiler-register!")
+    endif()
 endif()
\ No newline at end of file
diff --git a/samples/jpegDecodeBatched/CMakeLists.txt b/samples/jpegDecodeBatched/CMakeLists.txt
index 415650f1de..f14bb1317c 100644
--- a/samples/jpegDecodeBatched/CMakeLists.txt
+++ b/samples/jpegDecodeBatched/CMakeLists.txt
@@ -46,8 +46,9 @@ list(APPEND CMAKE_PREFIX_PATH ${ROCM_PATH}/hip ${ROCM_PATH})
 
 find_package(HIP QUIET)
 find_package(rocJPEG QUIET)
+find_package(rocprofiler-register QUIET)
 
-if(HIP_FOUND AND ROCJPEG_FOUND)
+if(HIP_FOUND AND ROCJPEG_FOUND AND rocprofiler-register_FOUND)
     # HIP
     set(LINK_LIBRARY_LIST ${LINK_LIBRARY_LIST} hip::host)
     # rocJPEG
@@ -55,6 +56,8 @@ if(HIP_FOUND AND ROCJPEG_FOUND)
     set(LINK_LIBRARY_LIST ${LINK_LIBRARY_LIST} ${ROCJPEG_LIBRARY})
     # std filesystem
     set(LINK_LIBRARY_LIST ${LINK_LIBRARY_LIST} stdc++fs)
+    # rocprofiler-register
+    set(LINK_LIBRARY_LIST ${LINK_LIBRARY_LIST} rocprofiler-register::rocprofiler-register)
 
     list(APPEND SOURCES ${PROJECT_SOURCE_DIR} jpegdecodebatched.cpp)
     add_executable(${PROJECT_NAME} ${SOURCES})
@@ -68,4 +71,7 @@ else()
     if (NOT ROCJPEG_FOUND)
         message(FATAL_ERROR "-- ERROR!: rocJPEG Not Found! - please install rocJPEG!")
     endif()
+    if (NOT rocprofiler-register_FOUND)
+        message(FATAL_ERROR "-- ERROR!: rocprofiler-register Not Found! - please install rocprofiler-register!")
+    endif()
 endif()
\ No newline at end of file
diff --git a/samples/jpegDecodePerf/CMakeLists.txt b/samples/jpegDecodePerf/CMakeLists.txt
index 49ca927d91..e61802aedc 100644
--- a/samples/jpegDecodePerf/CMakeLists.txt
+++ b/samples/jpegDecodePerf/CMakeLists.txt
@@ -46,11 +46,12 @@ list(APPEND CMAKE_PREFIX_PATH ${ROCM_PATH}/hip ${ROCM_PATH})
 
 find_package(HIP QUIET)
 find_package(rocJPEG QUIET)
+find_package(rocprofiler-register QUIET)
 
 # threads
 find_package(Threads REQUIRED)
 
-if(HIP_FOUND AND ROCJPEG_FOUND AND Threads_FOUND)
+if(HIP_FOUND AND ROCJPEG_FOUND AND Threads_FOUND AND rocprofiler-register_FOUND)
     # HIP
     set(LINK_LIBRARY_LIST ${LINK_LIBRARY_LIST} hip::host)
     #threads
@@ -58,6 +59,8 @@ if(HIP_FOUND AND ROCJPEG_FOUND AND Threads_FOUND)
     set(LINK_LIBRARY_LIST ${LINK_LIBRARY_LIST} Threads::Threads)
     # std filesystem
     set(LINK_LIBRARY_LIST ${LINK_LIBRARY_LIST} stdc++fs)
+    # rocprofiler-register
+    set(LINK_LIBRARY_LIST ${LINK_LIBRARY_LIST} rocprofiler-register::rocprofiler-register)
 
     # rocJPEG
     include_directories (${ROCJPEG_INCLUDE_DIR})
@@ -77,4 +80,7 @@ else()
     if (NOT Threads_FOUND)
         message(FATAL_ERROR "-- ERROR!: Threads Not Found! - please insatll Threads!")
     endif()
+    if (NOT rocprofiler-register_FOUND)
+        message(FATAL_ERROR "-- ERROR!: rocprofiler-register Not Found! - please install rocprofiler-register!")
+    endif()
 endif()
\ No newline at end of file