[dist] Add CUDA PTX architecutres to cv2.version

Use stable OpenCV and expand PTX architectures for build.

Author: Breakthrough

.github/workflows/build_wheels_windows.yml

@@ -24,6 +24,11 @@ on:
         required: false
         type: boolean
         default: true
+      rolling_build:
+        description: "Use latest commit from upstream OpenCV repo."
+        required: false
+        type: boolean
+        default: false
 
 
 jobs:
@@ -39,6 +44,7 @@ jobs:
       SDIST: 0
       ENABLE_HEADLESS: 0
       ENABLE_CONTRIB: 1
+      ENABLE_ROLLING: ${{ inputs.rolling_build && 1 || 0 }}
       OPENCV_TEST_DATA_PATH: ${{ github.workspace }}\opencv_extra\testdata
     steps:
     - name: Cleanup
@@ -47,12 +53,6 @@ jobs:
         rm -rf ./* || true
         rm -rf ./.??* || true
       working-directory: ${{ github.workspace }}
-    - name: Setup environment
-      shell: bash
-      run: |
-        if [[ "${{ github.event_name }}" == "schedule" || "${{ github.event_name }}" == "workflow_dispatch" ]]; then
-          echo "ENABLE_ROLLING=1" >> $GITHUB_ENV
-        fi
     - name: Checkout
       uses: actions/checkout@v4
       with:

find_version.py

@@ -7,6 +7,8 @@
     headless = sys.argv[2]
     rolling = sys.argv[3]
     ci_build = sys.argv[4]
+    cuda_arch_bin = sys.argv[5]
+    cuda_arch_ptx = sys.argv[6]
 
     opencv_version = ""
     # dig out the version from OpenCV sources
@@ -53,6 +55,12 @@
             print(e.output)
             raise
 
+
+    # TODO(@Breakthrough): Update this so that we just use the tag name directly as the version
+    # identifier. We might need to strip off a "v" suffix if we include that in the repo, but it
+    # will greatly simplify things. This can also get out of sync since we cache artifacts now.
+    # Move invoking this script to the builder!
+
     if len(tag) == 1:
         # tag identifies the build and should be a sequential revision number
         version = tag[0]
@@ -72,4 +80,5 @@
         f.write("contrib = {}\n".format(contrib))
         f.write("headless = {}\n".format(headless))
         f.write("rolling = {}\n".format(rolling))
-        f.write("ci_build = {}".format(ci_build))
+        f.write("cuda_arch_bin = {}".format(cuda_arch_bin))
+        f.write("cuda_arch_ptx = {}".format(cuda_arch_ptx))

setup.py

@@ -17,12 +17,24 @@ def main():
     CI_BUILD = os.environ.get("CI_BUILD", "False")
     is_CI_build = True if CI_BUILD == "1" else False
     cmake_source_dir = "opencv"
-    minimum_supported_numpy = "1.13.3"
     build_contrib = get_build_env_var_by_name("contrib")
     build_headless = get_build_env_var_by_name("headless")
     build_java = "ON" if get_build_env_var_by_name("java") else "OFF"
     build_rolling = get_build_env_var_by_name("rolling")
 
+    # fermi     = "2.0"
+    # kepler    = "3.0;3.5;3.7"
+    # maxwell   = "5.0;5.2"
+    # pascal    = "6.0;6.1"
+    # volta     = "7.0"
+    # turing    = "7.5"
+    # ampere    = "8.0;8.6"
+    # lovelace  = "8.9"
+    # hopper    = "9.0"
+    # blackwell =  "10.0;12.0"
+    cuda_arch_bin = "6.0;6.1;7.0;7.5,8.0"
+    cuda_arch_ptx = "8.0"
+
     # NOTE: since 2.3.0 numpy upgraded from manylinux2014 to manylinux_2_28
     # see https://numpy.org/doc/stable/release/2.3.0-notes.html#numpy-2-3-0-release-notes
     install_requires = [
@@ -70,7 +82,7 @@ def main():
                 )
 
     package_version, build_contrib, build_headless, build_rolling = get_and_set_info(
-        build_contrib, build_headless, build_rolling, is_CI_build
+        build_contrib, build_headless, build_rolling, is_CI_build, cuda_arch_bin, cuda_arch_ptx
     )
 
     # https://stackoverflow.com/questions/1405913/python-32bit-or-64bit-mode
@@ -151,8 +163,6 @@ def main():
 
     cudnn_library = os.environ["CUDNN_LIBRARY"]
     cudnn_include_dir = os.environ["CUDNN_INCLUDE_DIR"]
-    cuda_arch_bin = "6.0;6.1;7.0;7.5"
-    cuda_arch_ptx = "7.5"
 
     cmake_args = (
         (ci_cmake_generator if is_CI_build else [])
@@ -490,7 +500,7 @@ def _classify_installed_files_override(
         )
 
 
-def get_and_set_info(contrib, headless, rolling, ci_build):
+def get_and_set_info(contrib, headless, rolling, ci_build, cuda_arch_bin, cuda_arch_ptx):
     # cv2/version.py should be generated by running find_version.py
     version = {}
     here = os.path.abspath(os.path.dirname(__file__))
@@ -500,7 +510,7 @@ def get_and_set_info(contrib, headless, rolling, ci_build):
     # (in sdists the version.py file already exists)
     if os.path.exists(".git"):
         old_args = sys.argv.copy()
-        sys.argv = ["", str(contrib), str(headless), str(rolling), str(ci_build)]
+        sys.argv = ["", str(contrib), str(headless), str(rolling), str(ci_build), str(cuda_arch_bin), str(cuda_arch_ptx)]
         runpy.run_path("find_version.py", run_name="__main__")
         sys.argv = old_args