From 89a84131fbb4c50587f6d2d1d780b18986dbf7ab Mon Sep 17 00:00:00 2001
From: XSShawnZeng <shawnz@nvidia.com>
Date: Mon, 6 Jan 2025 18:47:18 +0800
Subject: [PATCH] Update CMakeLists.txt for Tegra sample simpleGLES

---
 .../Tegra/simpleGLES/CMakeLists.txt           | 44 +++++++++++++++----
 1 file changed, 36 insertions(+), 8 deletions(-)

diff --git a/Samples/8_Platform_Specific/Tegra/simpleGLES/CMakeLists.txt b/Samples/8_Platform_Specific/Tegra/simpleGLES/CMakeLists.txt
index 24e0b3fa..cc1bc80d 100644
--- a/Samples/8_Platform_Specific/Tegra/simpleGLES/CMakeLists.txt
+++ b/Samples/8_Platform_Specific/Tegra/simpleGLES/CMakeLists.txt
@@ -6,24 +6,52 @@ project(simpleGLES LANGUAGES C CXX CUDA)
 
 find_package(CUDAToolkit REQUIRED)
 
-set(CMAKE_CUDA_ARCHITECTURES 50 52 60 61 70 75 80 86 89 90)
+set(CMAKE_CUDA_ARCHITECTURES 53 61 70 72 75 80 86 87 90)
 if(CMAKE_BUILD_TYPE STREQUAL "Debug")
     # set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -G")  # enable cuda-gdb (expensive)
 endif()
 
 # Include directories and libraries
-include_directories(../../../Common)
+include_directories(../../../../Common)
+
+find_package(EGL)
+find_package(X11)
+find_package(OpenGL)
 
 if(CMAKE_SYSTEM_NAME STREQUAL "Linux")
     # Source file
-    # Add target for simpleGLES
-    add_executable(simpleGLES simpleGLES.cu)
+    if(${OpenGL_FOUND})
+        if(${EGL_FOUND})
+            if(${X11_FOUND})
+                # Add target for simpleGLES
+                add_executable(simpleGLES simpleGLES.cu)
 
-    target_compile_options(simpleGLES PRIVATE $<$<COMPILE_LANGUAGE:CUDA>:--extended-lambda>)
+                target_compile_options(simpleGLES PRIVATE $<$<COMPILE_LANGUAGE:CUDA>:--extended-lambda>)
 
-    target_compile_features(simpleGLES PRIVATE cxx_std_17 cuda_std_17)
+                target_compile_features(simpleGLES PRIVATE cxx_std_17 cuda_std_17)
 
-    set_target_properties(simpleGLES PROPERTIES CUDA_SEPARABLE_COMPILATION ON)
+                set_target_properties(simpleGLES PROPERTIES CUDA_SEPARABLE_COMPILATION ON)
+
+                target_include_directories(simpleGLES PUBLIC
+                    ${EGL_INCLUDE_DIR}
+                    ${OPENGL_INCLUDE_DIR}
+                    ${CUDAToolkit_INCLUDE_DIRS}
+                )
+
+                target_link_libraries(simpleGLES
+                    ${EGL_LIBRARY}
+                    ${X11_LIBRARIES}
+                    ${OPENGL_LIBRARIES}
+                )
+            else()
+                message(STATUS "X11 libraries not found - will not build sample 'simpleGLES'")
+            endif()
+        else()
+            message(STATUS "EGL not found - will not build sample 'simpleGLES'")
+        endif()
+    else()
+        message(STATUS "OpenGL not found - will not build sample 'simpleGLES'")
+    endif()
 else()
     message(STATUS "Will not build sample simpleGLES - requires Linux OS")
-endif()
\ No newline at end of file
+endif()