Merge pull request #9 from koide3/orin

Orin
koide3 · Jul 22, 2024 · 1c07055 · 1c07055
2 parents b3a7d04 + f63e1f8
commit 1c07055
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -45,16 +45,16 @@ if(BUILD_WITH_CUDA)
   if(${CMAKE_VERSION} VERSION_GREATER_EQUAL "3.24.0")
     if(BUILD_WITH_CUDA_MULTIARCH)
       set(CMAKE_CUDA_ARCHITECTURES "all-major" CACHE STRING "CUDA architectures" FORCE)
-    else()
+    elseif(NOT DEFINED CMAKE_CUDA_ARCHITECTURES OR CMAKE_CUDA_ARCHITECTURES STREQUAL "52")
       set(CMAKE_CUDA_ARCHITECTURES "native" CACHE STRING "CUDA architectures" FORCE)
     endif()
   else()
     if(BUILD_WITH_CUDA_MULTIARCH)
-      set(CMAKE_CUDA_ARCHITECTURES 75 80 89 90 CACHE STRING "CUDA architectures" FORCE)
+      set(CMAKE_CUDA_ARCHITECTURES 75 80 87 89 90 CACHE STRING "CUDA architectures" FORCE)
     else()
       if(NOT DEFINED CMAKE_CUDA_ARCHITECTURES OR CMAKE_CUDA_ARCHITECTURES STREQUAL "52")
-        message(WARNING "CMAKE_CUDA_ARCHITECTURES is not defined. Set to 89 by default")
-        set(CMAKE_CUDA_ARCHITECTURES 89 CACHE STRING "CUDA architectures" FORCE)
+        message(WARNING "CMAKE_CUDA_ARCHITECTURES is not defined. Set to 80 by default")
+        set(CMAKE_CUDA_ARCHITECTURES 80 CACHE STRING "CUDA architectures" FORCE)
       endif()
     endif()
   endif()

diff --git a/src/gtsam_points/cuda/gl_buffer_map.cu b/src/gtsam_points/cuda/gl_buffer_map.cu
@@ -3,6 +3,7 @@
 
 #include <gtsam_points/cuda/gl_buffer_map.hpp>
 
+#include <GL/gl.h>
 #include <cuda_runtime.h>
 #include <cuda_gl_interop.h>