From 358e3065abe9caa5ab7b86e004c85b90709f8ecf Mon Sep 17 00:00:00 2001
From: cudawarped <12133430+cudawarped@users.noreply.github.com>
Date: Wed, 9 Aug 2023 09:21:11 +0300
Subject: [PATCH] cuda: add default ptx when CUDA_ARCH_BIN is missing and allow
 CUDA_ARCH_PTX to be passed in isolation

---
 cmake/OpenCVDetectCUDA.cmake | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/cmake/OpenCVDetectCUDA.cmake b/cmake/OpenCVDetectCUDA.cmake
index 4a562bdaf9..752ae0edcf 100644
--- a/cmake/OpenCVDetectCUDA.cmake
+++ b/cmake/OpenCVDetectCUDA.cmake
@@ -228,7 +228,7 @@ if(CUDA_FOUND)
     endif()
   endmacro()
 
-  set(__cuda_arch_ptx "")
+  set(__cuda_arch_ptx ${CUDA_ARCH_PTX})
   if(CUDA_GENERATION STREQUAL "Fermi")
     set(__cuda_arch_bin ${_arch_fermi})
   elseif(CUDA_GENERATION STREQUAL "Kepler")
@@ -259,7 +259,7 @@ if(CUDA_FOUND)
     set(__cuda_arch_bin ${CUDA_ARCH_BIN})
   endif()
 
-  if(NOT DEFINED __cuda_arch_bin)
+  if(NOT DEFINED __cuda_arch_bin AND NOT DEFINED __cuda_arch_ptx)
     if(ARM)
       set(__cuda_arch_bin "3.2")
       set(__cuda_arch_ptx "")
@@ -295,6 +295,7 @@ if(CUDA_FOUND)
           ${_arch_lovelace}
           ${_arch_hopper}
       )
+      list(GET __cuda_arch_bin -1 __cuda_arch_ptx)
     endif()
   endif()