PyNvVideoCodec 2.2.1__cp313-cp313-win_arm64.whl

benchmarks/setup_pycuda_win_arm64.bat

@@ -0,0 +1,531 @@
+@echo off
+rem This copyright notice applies to this file only
+
+rem SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+rem SPDX-License-Identifier: MIT
+
+rem Permission is hereby granted, free of charge, to any person obtaining a
+rem copy of this software and associated documentation files (the "Software"),
+rem to deal in the Software without restriction, including without limitation
+rem the rights to use, copy, modify, merge, publish, distribute, sublicense,
+rem and/or sell copies of the Software, and to permit persons to whom the
+rem Software is furnished to do so, subject to the following conditions:
+
+rem The above copyright notice and this permission notice shall be included in
+rem all copies or substantial portions of the Software.
+
+rem THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+rem IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+rem FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
+rem THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+rem LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+rem FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
+rem DEALINGS IN THE SOFTWARE.
+
+setlocal EnableExtensions EnableDelayedExpansion
+
+rem Wheels or source checkouts can preserve this file with LF-only line endings.
+rem cmd.exe may then fail to resolve later "call :label" subroutines. Relaunch
+rem once from a CRLF-normalized temp copy before using any subroutine calls.
+if not defined PYNVC_PYCUDA_CRLF_NORMALIZED (
+    set "PYNVC_PYCUDA_CRLF_NORMALIZED=1"
+    set "PYNVC_PYCUDA_ORIGINAL_BAT=%~f0"
+    set "PYNVC_PYCUDA_CRLF_BAT=%TEMP%\%~n0_%RANDOM%_%RANDOM%.bat"
+    powershell.exe -NoProfile -ExecutionPolicy Bypass -Command "$src=$env:PYNVC_PYCUDA_ORIGINAL_BAT; $dst=$env:PYNVC_PYCUDA_CRLF_BAT; $text=[System.IO.File]::ReadAllText($src); $text=[System.Text.RegularExpressions.Regex]::Replace($text, '\r?\n', [Environment]::NewLine); [System.IO.File]::WriteAllText($dst, $text, [System.Text.Encoding]::ASCII)"
+    if errorlevel 1 (
+        echo ERROR: Failed to create a CRLF-normalized copy of "%~f0".
+        exit /b 1
+    )
+    call "!PYNVC_PYCUDA_CRLF_BAT!" %*
+    set "PYNVC_PYCUDA_CRLF_STATUS=!ERRORLEVEL!"
+    del /q "!PYNVC_PYCUDA_CRLF_BAT!" >nul 2>nul
+    exit /b !PYNVC_PYCUDA_CRLF_STATUS!
+)
+set "PYCUDA_VERSION=2026.1"
+set "PYCUDA_TAG=v2026.1"
+set "PYCUDA_COMMIT=7610751702a013053fab8b604bdf1c236dd1f3c0"
+set "PYCUDA_REPO=https://github.com/inducer/pycuda.git"
+set "BPL_SUBSET_COMMIT=021f0d90d2730cc138b4dd476cb3184b27c2a33e"
+set "COMPYTE_COMMIT=f1010ef64b5d4559ca4bb7690d4b3394091477b6"
+set "PYTOOLS_VERSION=2024.1.14"
+
+if not defined PYTHON_EXE set "PYTHON_EXE=python"
+if not defined PYCUDA_BUILD_ROOT set "PYCUDA_BUILD_ROOT=%LOCALAPPDATA%\PyNvVideoCodec\pycuda-win-arm64\%PYCUDA_VERSION%"
+if not defined PYCUDA_SOURCE_DIR set "PYCUDA_SOURCE_DIR=%PYCUDA_BUILD_ROOT%\src"
+
+echo.
+echo PyNvVideoCodec PyCUDA setup for native Windows ARM64
+echo =====================================================
+echo.
+
+call :check_python
+if errorlevel 1 goto :fail
+
+call :activate_msvc
+if errorlevel 1 goto :fail
+
+call :find_cuda
+if errorlevel 1 goto :fail
+
+call :install_python_build_deps
+if errorlevel 1 goto :fail
+
+call :sync_pycuda_source
+if errorlevel 1 goto :fail
+
+call :write_siteconf
+if errorlevel 1 goto :fail
+
+call :install_pycuda
+if errorlevel 1 goto :fail
+
+call :install_cuda_dll_path_hook
+if errorlevel 1 goto :fail
+
+call :verify_pycuda
+if errorlevel 1 goto :fail
+
+echo.
+echo PyCUDA setup completed successfully.
+exit /b 0
+
+:check_python
+echo Checking Python...
+"%PYTHON_EXE%" -c "import platform, sys; print('  executable:', sys.executable); print('  version   :', sys.version.split()[0]); print('  machine   :', platform.machine()); raise SystemExit(0 if sys.implementation.name == 'cpython' and sys.maxsize > 2**32 and platform.machine().lower() in ('arm64', 'aarch64') else 1)"
+if errorlevel 1 (
+    echo ERROR: PYTHON_EXE must point to a native 64-bit CPython ARM64 interpreter.
+    echo        Set PYTHON_EXE to the ARM64 python.exe in your target environment and rerun this script.
+    exit /b 1
+)
+exit /b 0
+
+:activate_msvc
+if /I "%VSCMD_ARG_TGT_ARCH%"=="arm64" (
+    where cl.exe >nul 2>nul
+    if not errorlevel 1 (
+        where link.exe >nul 2>nul
+        if not errorlevel 1 (
+            echo Using active MSVC environment for target ARM64.
+            goto :msvc_verify
+        )
+    )
+    echo Existing MSVC target is ARM64, but cl.exe/link.exe are not on PATH.
+    echo Re-activating MSVC from vcvarsall.bat...
+)
+
+call :find_vcvarsall
+if errorlevel 1 exit /b 1
+
+echo Activating MSVC ARM64 build environment...
+call "%VCVARSALL%" arm64
+if errorlevel 1 (
+    echo ERROR: vcvarsall.bat failed for target arm64.
+    exit /b 1
+)
+
+:msvc_verify
+if /I not "%VSCMD_ARG_TGT_ARCH%"=="arm64" (
+    echo ERROR: MSVC target architecture is "%VSCMD_ARG_TGT_ARCH%", expected "arm64".
+    exit /b 1
+)
+
+where cl.exe >nul 2>nul
+if errorlevel 1 (
+    echo ERROR: cl.exe was not found after activating Visual Studio.
+    echo        Make sure this Visual Studio installation has the MSVC ARM64 C++ build tools installed.
+    exit /b 1
+)
+
+where link.exe >nul 2>nul
+if errorlevel 1 (
+    echo ERROR: link.exe was not found after activating Visual Studio.
+    echo        Make sure this Visual Studio installation has the MSVC ARM64 C++ build tools installed.
+    exit /b 1
+)
+
+echo MSVC target: %VSCMD_ARG_TGT_ARCH%
+exit /b 0
+
+:find_vcvarsall
+if defined VCVARSALL (
+    if exist "%VCVARSALL%" exit /b 0
+
+    echo ERROR: VCVARSALL is set but does not exist:
+    echo        "%VCVARSALL%"
+    echo        Update VCVARSALL or unset it so this script can auto-detect Visual Studio.
+    exit /b 1
+)
+
+if defined VSINSTALLDIR (
+    if exist "%VSINSTALLDIR%\VC\Auxiliary\Build\vcvarsall.bat" (
+        set "VCVARSALL=%VSINSTALLDIR%\VC\Auxiliary\Build\vcvarsall.bat"
+        exit /b 0
+    )
+)
+
+set "VSWHERE="
+if exist "%ProgramFiles(x86)%\Microsoft Visual Studio\Installer\vswhere.exe" (
+    set "VSWHERE=%ProgramFiles(x86)%\Microsoft Visual Studio\Installer\vswhere.exe"
+) else if exist "%ProgramFiles%\Microsoft Visual Studio\Installer\vswhere.exe" (
+    set "VSWHERE=%ProgramFiles%\Microsoft Visual Studio\Installer\vswhere.exe"
+) else (
+    for %%I in (vswhere.exe) do set "VSWHERE=%%~$PATH:I"
+)
+
+if defined VSWHERE (
+    for /f "usebackq delims=" %%I in (`"%VSWHERE%" -latest -products * -requires Microsoft.VisualStudio.Component.VC.Tools.ARM64 -property installationPath`) do (
+        if exist "%%I\VC\Auxiliary\Build\vcvarsall.bat" (
+            set "VCVARSALL=%%I\VC\Auxiliary\Build\vcvarsall.bat"
+            exit /b 0
+        )
+    )
+)
+
+echo ERROR: Could not find Visual Studio vcvarsall.bat with ARM64 C++ tools.
+echo        Install the C++ ARM64 build tools, or set VCVARSALL before running this script.
+echo        Example:
+echo        set "VCVARSALL=^<Visual Studio root^>\VC\Auxiliary\Build\vcvarsall.bat"
+exit /b 1
+
+
+:find_cuda
+echo Checking CUDA Toolkit...
+if not defined CUDA_PATH (
+    for /f "delims=" %%I in ('where nvcc.exe 2^>nul') do (
+        for %%J in ("%%~dpI..") do set "CUDA_PATH=%%~fJ"
+        goto :cuda_path_found
+    )
+)
+
+:cuda_path_found
+if not defined CUDA_PATH (
+    echo ERROR: CUDA_PATH is not set and nvcc.exe was not found on PATH.
+    echo        Install the native Windows ARM64 CUDA Toolkit, or set CUDA_PATH before running this script.
+    exit /b 1
+)
+
+if not exist "%CUDA_PATH%\include\cuda.h" (
+    echo ERROR: "%CUDA_PATH%\include\cuda.h" was not found.
+    echo        CUDA_PATH must point to the CUDA Toolkit root.
+    exit /b 1
+)
+
+if not exist "%CUDA_PATH%\lib\arm64\cuda.lib" (
+    echo ERROR: "%CUDA_PATH%\lib\arm64\cuda.lib" was not found.
+    echo        This script only builds native ARM64 PyCUDA and will not link against x64 CUDA libraries.
+    exit /b 1
+)
+
+set "CUDA_LIB_DIR=%CUDA_PATH%\lib\arm64"
+set "CUDA_ENABLE_CURAND=False"
+if exist "%CUDA_LIB_DIR%\curand.lib" set "CUDA_ENABLE_CURAND=True"
+
+rem CUDA 13+ Windows ARM64 toolkits keep runtime DLLs under bin\arm64 (and
+rem x64 under bin\x64). Prefer the native arm64 runtime dir so Python does not
+rem pick up an x64 curand/cudart DLL ("not a valid Win32 application").
+set "CUDA_BIN_DIR="
+if exist "%CUDA_PATH%\bin\arm64\curand64_10.dll" (
+    set "CUDA_BIN_DIR=%CUDA_PATH%\bin\arm64"
+) else if exist "%CUDA_PATH%\bin\arm64\cudart64_13.dll" (
+    set "CUDA_BIN_DIR=%CUDA_PATH%\bin\arm64"
+) else if exist "%CUDA_PATH%\bin\curand64_10.dll" (
+    set "CUDA_BIN_DIR=%CUDA_PATH%\bin"
+) else if exist "%CUDA_PATH%\bin\cudart64_13.dll" (
+    set "CUDA_BIN_DIR=%CUDA_PATH%\bin"
+) else if exist "%CUDA_PATH%\bin" (
+    set "CUDA_BIN_DIR=%CUDA_PATH%\bin"
+)
+
+if not defined CUDA_BIN_DIR (
+    echo ERROR: Could not locate a CUDA runtime bin directory under "%CUDA_PATH%\bin".
+    exit /b 1
+)
+
+set "PATH=%CUDA_BIN_DIR%;%CUDA_PATH%\bin;%PATH%"
+
+if not exist "%CUDA_PATH%\bin\nvcc.exe" (
+    echo WARNING: nvcc.exe was not found under CUDA_PATH\bin. PyCUDA driver APIs can still build, but pycuda.compiler will need nvcc later.
+)
+
+echo CUDA root : %CUDA_PATH%
+echo CUDA libs : %CUDA_LIB_DIR%
+echo CUDA bins : %CUDA_BIN_DIR%
+echo CURAND    : %CUDA_ENABLE_CURAND%
+exit /b 0
+
+:install_python_build_deps
+echo Installing Python build dependencies...
+"%PYTHON_EXE%" -m pip install --upgrade --prefer-binary setuptools wheel "numpy>=1.24" mako "platformdirs>=2.2.0" "pytools==%PYTOOLS_VERSION%"
+if errorlevel 1 (
+    echo ERROR: Failed to install Python build dependencies.
+    exit /b 1
+)
+exit /b 0
+
+:sync_pycuda_source
+echo Preparing PyCUDA source in "%PYCUDA_SOURCE_DIR%"...
+if not exist "%PYCUDA_BUILD_ROOT%" mkdir "%PYCUDA_BUILD_ROOT%"
+if errorlevel 1 (
+    echo ERROR: Could not create "%PYCUDA_BUILD_ROOT%".
+    exit /b 1
+)
+
+where git.exe >nul 2>nul
+if errorlevel 1 (
+    echo git.exe was not found on PATH. Using GitHub ZIP download fallback.
+    call :download_pycuda_source_without_git
+    exit /b !ERRORLEVEL!
+)
+
+if exist "%PYCUDA_SOURCE_DIR%\.git" (
+    git -C "%PYCUDA_SOURCE_DIR%" fetch --tags --depth 1 origin "%PYCUDA_TAG%"
+    if errorlevel 1 (
+        echo WARNING: Git fetch failed. Trying GitHub ZIP download fallback.
+        call :download_pycuda_source_without_git
+        exit /b !ERRORLEVEL!
+    )
+    git -C "%PYCUDA_SOURCE_DIR%" checkout --detach "%PYCUDA_TAG%"
+    if errorlevel 1 (
+        echo WARNING: Git checkout failed. Trying GitHub ZIP download fallback.
+        call :download_pycuda_source_without_git
+        exit /b !ERRORLEVEL!
+    )
+) else (
+    git clone --branch "%PYCUDA_TAG%" --depth 1 --recurse-submodules "%PYCUDA_REPO%" "%PYCUDA_SOURCE_DIR%"
+    if errorlevel 1 (
+        echo WARNING: Git clone failed. Trying GitHub ZIP download fallback.
+        call :download_pycuda_source_without_git
+        exit /b !ERRORLEVEL!
+    )
+)
+
+git -C "%PYCUDA_SOURCE_DIR%" submodule update --init --recursive
+if errorlevel 1 (
+    echo WARNING: Git submodule update failed. Trying GitHub ZIP download fallback.
+    call :download_pycuda_source_without_git
+    exit /b !ERRORLEVEL!
+)
+exit /b 0
+
+:download_pycuda_source_without_git
+where curl.exe >nul 2>nul
+if errorlevel 1 (
+    echo ERROR: git.exe is unavailable and curl.exe was not found.
+    echo        Install Git for Windows, or make curl.exe available on PATH.
+    exit /b 1
+)
+
+where tar.exe >nul 2>nul
+if errorlevel 1 (
+    echo ERROR: git.exe is unavailable and tar.exe was not found.
+    echo        Install Git for Windows, or make tar.exe available on PATH.
+    exit /b 1
+)
+
+call :download_and_extract_zip "https://github.com/inducer/pycuda/archive/%PYCUDA_COMMIT%.zip" "%PYCUDA_SOURCE_DIR%" "pycuda-"
+if errorlevel 1 exit /b 1
+
+call :download_and_extract_zip "https://github.com/inducer/bpl-subset/archive/%BPL_SUBSET_COMMIT%.zip" "%PYCUDA_SOURCE_DIR%\bpl-subset" "bpl-subset-"
+if errorlevel 1 exit /b 1
+
+call :download_and_extract_zip "https://github.com/inducer/compyte/archive/%COMPYTE_COMMIT%.zip" "%PYCUDA_SOURCE_DIR%\pycuda\compyte" "compyte-"
+if errorlevel 1 exit /b 1
+
+if not exist "%PYCUDA_SOURCE_DIR%\setup.py" (
+    echo ERROR: PyCUDA setup.py was not found after ZIP download.
+    exit /b 1
+)
+
+if not exist "%PYCUDA_SOURCE_DIR%\bpl-subset\bpl_subset\boost\python.hpp" (
+    echo ERROR: bpl-subset was not expanded correctly.
+    exit /b 1
+)
+
+if not exist "%PYCUDA_SOURCE_DIR%\pycuda\compyte\array.py" (
+    echo ERROR: compyte was not expanded correctly.
+    exit /b 1
+)
+
+exit /b 0
+
+:download_and_extract_zip
+set "ZIP_URL=%~1"
+set "ZIP_DEST=%~2"
+set "ZIP_PREFIX=%~3"
+set "ZIP_TEMP=%PYCUDA_BUILD_ROOT%\_zip_%RANDOM%_%RANDOM%"
+set "ZIP_FILE=%ZIP_TEMP%\source.zip"
+set "ZIP_EXTRACT_DIR=%ZIP_TEMP%\extract"
+set "ZIP_EXTRACTED="
+
+echo Downloading %ZIP_URL%
+if not exist "%ZIP_TEMP%" mkdir "%ZIP_TEMP%"
+if errorlevel 1 exit /b 1
+mkdir "%ZIP_EXTRACT_DIR%"
+if errorlevel 1 exit /b 1
+
+curl.exe -L --fail --retry 3 -o "%ZIP_FILE%" "%ZIP_URL%"
+if errorlevel 1 (
+    echo ERROR: Failed to download %ZIP_URL%
+    exit /b 1
+)
+
+tar.exe -xf "%ZIP_FILE%" -C "%ZIP_EXTRACT_DIR%"
+if errorlevel 1 (
+    echo ERROR: Failed to extract "%ZIP_FILE%".
+    exit /b 1
+)
+
+for /d %%D in ("%ZIP_EXTRACT_DIR%\%ZIP_PREFIX%*") do set "ZIP_EXTRACTED=%%~fD"
+if not defined ZIP_EXTRACTED (
+    echo ERROR: Could not find extracted folder matching "%ZIP_PREFIX%*" from %ZIP_URL%
+    exit /b 1
+)
+
+if not exist "%ZIP_DEST%" mkdir "%ZIP_DEST%"
+if errorlevel 1 exit /b 1
+
+robocopy "%ZIP_EXTRACTED%" "%ZIP_DEST%" /E >nul
+set "ROBOCOPY_STATUS=!ERRORLEVEL!"
+
+rmdir /s /q "%ZIP_TEMP%" >nul 2>nul
+
+if !ROBOCOPY_STATUS! GEQ 8 (
+    echo ERROR: Failed to copy extracted files into "%ZIP_DEST%".
+    exit /b !ROBOCOPY_STATUS!
+)
+
+exit /b 0
+
+:write_siteconf
+echo Writing PyCUDA siteconf.py...
+(
+    echo # Generated by setup_pycuda_win_arm64.bat.
+    echo BOOST_INC_DIR = []
+    echo BOOST_LIB_DIR = []
+    echo BOOST_COMPILER = "msvc"
+    echo USE_SHIPPED_BOOST = True
+    echo BOOST_PYTHON_LIBNAME = []
+    echo BOOST_THREAD_LIBNAME = []
+    echo CUDA_TRACE = False
+    echo CUDA_ROOT = r"%CUDA_PATH%"
+    echo CUDA_PRETEND_VERSION = None
+    echo CUDA_INC_DIR = [r"%CUDA_PATH%\include"]
+    echo CUDA_ENABLE_GL = False
+    echo CUDA_ENABLE_CURAND = %CUDA_ENABLE_CURAND%
+    echo CUDADRV_LIB_DIR = [r"%CUDA_LIB_DIR%"]
+    echo CUDADRV_LIBNAME = ["cuda"]
+    echo CUDART_LIB_DIR = [r"%CUDA_LIB_DIR%"]
+    echo CUDART_LIBNAME = ["cudart"]
+    echo CURAND_LIB_DIR = [r"%CUDA_LIB_DIR%"]
+    echo CURAND_LIBNAME = ["curand"]
+    echo CXXFLAGS = ["/EHsc", "/D_LITTLE_ENDIAN=1"]
+    echo LDFLAGS = ["/FORCE", "/MANIFEST:NO"]
+) > "%PYCUDA_SOURCE_DIR%\siteconf.py"
+if errorlevel 1 (
+    echo ERROR: Failed to write PyCUDA siteconf.py.
+    exit /b 1
+)
+exit /b 0
+
+:install_pycuda
+echo Building and installing PyCUDA %PYCUDA_VERSION%...
+pushd "%PYCUDA_SOURCE_DIR%"
+if errorlevel 1 exit /b 1
+
+set "DISTUTILS_USE_SDK=1"
+set "MSSdk=1"
+set "CL=/D_LITTLE_ENDIAN=1 %CL%"
+
+"%PYTHON_EXE%" -m pip install --no-build-isolation --no-deps --no-cache-dir --force-reinstall -v .
+set "PIP_STATUS=%ERRORLEVEL%"
+popd
+
+if not "%PIP_STATUS%"=="0" (
+    echo ERROR: PyCUDA build/install failed.
+    exit /b %PIP_STATUS%
+)
+exit /b 0
+
+:install_cuda_dll_path_hook
+rem Python 3.8+ on Windows does not search PATH for extension-module DLL
+rem dependencies. Register the native CUDA runtime dir via a site hook so
+rem "import pycuda" finds curand/cudart without manual add_dll_directory.
+echo Installing CUDA DLL search path hook for "%CUDA_BIN_DIR%"...
+
+if not defined CUDA_BIN_DIR (
+    echo ERROR: CUDA_BIN_DIR is not set.
+    exit /b 1
+)
+
+if not exist "%CUDA_BIN_DIR%" (
+    echo ERROR: CUDA_BIN_DIR does not exist: "%CUDA_BIN_DIR%"
+    exit /b 1
+)
+
+set "PYCUDA_SITE_DIR="
+set "PYCUDA_SITE_FILE=%PYCUDA_BUILD_ROOT%\pycuda_site_dir.txt"
+"%PYTHON_EXE%" -c "import sysconfig; print(sysconfig.get_path('purelib'))" > "%PYCUDA_SITE_FILE%"
+if errorlevel 1 (
+    echo ERROR: Could not query Python site-packages directory.
+    exit /b 1
+)
+set /p PYCUDA_SITE_DIR=<"%PYCUDA_SITE_FILE%"
+del /q "%PYCUDA_SITE_FILE%" >nul 2>nul
+if not defined PYCUDA_SITE_DIR (
+    echo ERROR: Could not determine Python site-packages directory.
+    exit /b 1
+)
+
+if not exist "%PYCUDA_SITE_DIR%" (
+    echo ERROR: Python site-packages directory does not exist: "%PYCUDA_SITE_DIR%"
+    exit /b 1
+)
+
+set "PYCUDA_DLL_HOOK=%PYCUDA_SITE_DIR%\_pycuda_cuda_dll_path.py"
+set "PYCUDA_DLL_PTH=%PYCUDA_SITE_DIR%\_pycuda_cuda_dll_path.pth"
+
+> "%PYCUDA_DLL_HOOK%" echo import os
+>> "%PYCUDA_DLL_HOOK%" echo from pathlib import Path
+>> "%PYCUDA_DLL_HOOK%" echo(
+>> "%PYCUDA_DLL_HOOK%" echo _CUDA_BIN = Path(r'''%CUDA_BIN_DIR%''')
+>> "%PYCUDA_DLL_HOOK%" echo _DLL_DIR_HANDLES = []
+>> "%PYCUDA_DLL_HOOK%" echo(
+>> "%PYCUDA_DLL_HOOK%" echo def _add():
+>> "%PYCUDA_DLL_HOOK%" echo     if not _CUDA_BIN.is_dir():
+>> "%PYCUDA_DLL_HOOK%" echo         return
+>> "%PYCUDA_DLL_HOOK%" echo     if hasattr(os, "add_dll_directory"):
+>> "%PYCUDA_DLL_HOOK%" echo         _DLL_DIR_HANDLES.append(os.add_dll_directory(str(_CUDA_BIN)))
+>> "%PYCUDA_DLL_HOOK%" echo     path = os.environ.get("PATH", "")
+>> "%PYCUDA_DLL_HOOK%" echo     prefix = str(_CUDA_BIN)
+>> "%PYCUDA_DLL_HOOK%" echo     parts = [p.lower() for p in path.split(os.pathsep)]
+>> "%PYCUDA_DLL_HOOK%" echo     if prefix.lower() not in parts:
+>> "%PYCUDA_DLL_HOOK%" echo         os.environ["PATH"] = prefix + os.pathsep + path
+>> "%PYCUDA_DLL_HOOK%" echo(
+>> "%PYCUDA_DLL_HOOK%" echo _add()
+
+> "%PYCUDA_DLL_PTH%" echo import _pycuda_cuda_dll_path
+
+if errorlevel 1 (
+    echo ERROR: Failed to install CUDA DLL search path hook.
+    exit /b 1
+)
+
+echo   site: %PYCUDA_SITE_DIR%
+echo   cuda bin: %CUDA_BIN_DIR%
+echo   wrote: _pycuda_cuda_dll_path.py, _pycuda_cuda_dll_path.pth
+exit /b 0
+
+:verify_pycuda
+echo Verifying PyCUDA import and CUDA driver initialization...
+"%PYTHON_EXE%" -u -X faulthandler -c "import platform, pycuda, pycuda.driver as cuda; print('  pycuda:', pycuda.VERSION_TEXT); print('  machine:', platform.machine()); cuda.init(); print('  cuda.init: ok'); print('  devices:', cuda.Device.count())"
+if errorlevel 1 (
+    echo ERROR: PyCUDA verification failed.
+    echo        If import still fails, ensure CUDA runtime DLLs are under "%CUDA_BIN_DIR%".
+    exit /b 1
+)
+exit /b 0
+
+:fail
+echo.
+echo PyCUDA setup failed.
+exit /b 1

pynvvideocodec-2.2.1.data/data/external/ffmpeg/readme.txt

@@ -0,0 +1,39 @@
+FFMPEG Libraries present in this package were built using the following configure options
+
+win32:  ./configure --enable-shared --disable-encoders --disable-decoders --enable-decoder=vp9 --toolchain=msvc --arch=x86
+aarch64: LDSOFLAGS=-Wl,-rpath,\''$$$$ORIGIN'\'  ./configure --enable-shared --disable-encoders --disable-decoders --enable-decoder=vp9  --arch=aarch64
+
+x64 (FFmpeg 8.1.2, demux/mux only):
+    ./configure \
+        --prefix=<prefix> --toolchain=msvc \
+        --enable-static --disable-shared --enable-pic --disable-x86asm \
+        --disable-everything --disable-programs --disable-doc --disable-libdrm \
+        --disable-encoders --disable-decoders --disable-hwaccels \
+        --disable-swresample --disable-swscale --disable-avfilter --disable-avdevice \
+        --enable-demuxers --enable-muxers --enable-protocols --enable-parsers --enable-bsfs \
+        --enable-avformat --enable-avcodec --enable-avutil
+
+    After building the static archives, avutil-60.dll and avformat-62.dll were
+    produced from the .lib archives via `link.exe -DLL -WHOLEARCHIVE:...` (the MSVC
+    analog of `gcc -shared -Wl,--whole-archive`). libavcodec is STATICALLY EMBEDDED
+    inside avformat-62.dll (no separate avcodec-*.dll is shipped). avutil.def is
+    extended via `dumpbin -symbols libavutil.lib` to also export internal ff_*
+    symbols, since MSVC otherwise hides them.
+
+x86_64 (FFmpeg 8.1.2, demux/mux only):
+    ./configure \
+        --prefix=<prefix> \
+        --enable-static --disable-shared --enable-pic --disable-x86asm \
+        --disable-everything --disable-programs --disable-doc --disable-libdrm \
+        --disable-encoders --disable-decoders --disable-hwaccels \
+        --disable-swresample --disable-swscale --disable-avfilter --disable-avdevice \
+        --enable-demuxers --enable-muxers --enable-protocols --enable-parsers --enable-bsfs \
+        --enable-avformat --enable-avcodec --enable-avutil
+
+    After building the static archives, libavutil.so.60 and libavformat.so.62 were
+    produced from the .a archives via `gcc -shared -Wl,--whole-archive ...`. libavcodec
+    is STATICALLY EMBEDDED inside libavformat.so (no separate libavcodec.so is shipped).
+    Only libavformat / libavutil headers + libavcodec headers are provided; swresample,
+    swscale, avfilter, avdevice are not built into this distribution.
+
+Only header files and libraries that are required for compiling the sample applications inside the package have been included.

pynvvideocodec-2.2.1.dist-info/METADATA

@@ -0,0 +1,97 @@
+Metadata-Version: 2.4
+Name: pynvvideocodec
+Version: 2.2.1
+Summary: pynvvideocodec (PyNvVideoCodec) is NVIDIA's Python library for hardware-accelerated video encode/decode on NVIDIA GPUs.
+Author: NVIDIA
+Maintainer: nvidia
+License: MIT
+Project-URL: documentation, https://docs.nvidia.com/video-technologies/pynvvideocodec/index.html
+Project-URL: source, https://catalog.ngc.nvidia.com/orgs/nvidia/resources/pynvvideocodec
+Classifier: License :: OSI Approved :: MIT License
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE.txt
+License-File: NOTICES.txt
+Dynamic: license-file
+
+# **PyNvVideoCodec**
+
+[PyNvVideoCodec](https://docs.nvidia.com/video-technologies/pynvvideocodec/index.html) is NVIDIA's Python-based library that provides simple yet powerful Python APIs for hardware-accelerated video encoding and decoding on NVIDIA GPUs.
+
+PyNvVideoCodec is built on top of the Video Codec SDK and offers encode, decode, and transcode performance on par with it.
+
+The library is distributed under MIT license and is officially supported by NVIDIA.
+
+## **Features**
+
+PyNvVideoCodec supports following features:
+
+### **Decode Features**
+
+* **Seek and frame sampling:** Provides efficient and flexible methods for fetching video frames in various modes, including sequential, random, periodic, indexed, batched, and sliced, as well as at a specified target frame rate.
+* **Decoder caching:** Optimizes decoding of short video clips through decoder caching and reconfiguration.
+* **Threaded decoder:** Supports decoding on separate threads, delivering pre-decoded frames with near-zero latency, enabling high-performance video processing pipelines.
+* **Decode statistics extraction:** Enables extraction of low-level decode statistics—such as QP values, coding-unit types, and motion vectors—from H.264 and H.265 streams. These statistics provide valuable insights into video quality, content complexity, and encoder behaviour.
+* **Video processing from buffer:** Supports video processing from memory buffers, reducing I/O overhead, enabling streaming applications.
+* **Low latency decode:** Offers zero-latency decoding for video sequences that do not contain B-frames.
+* **SEI extraction:** Supports the extraction of Supplemental Enhancement Information (SEI) messages, allowing access to additional information such as HDR information, timecodes, and custom user data.
+* **Stream metadata access:** Enables access to stream metadata, including frame width, height, bit depth, and keyframe indices, to enhance content management.
+* **GIL handling:** Improved multithreaded performance through better handling of Global Interpreter Lock (GIL) in C++ layer.
+* **Multi-GPU decode:** Enables multi-GPU decoding to efficiently handle larger workloads.
+* **Extended codec support:** Supports codecs H.264, HEVC, AV1, VP8, VP9, VC1, MPEG4, MPEG2, and MPEG1
+* **4:2:2 decode:** Supports 4:2:2 decoding for both H.264 and HEVC formats on Blackwell GPUs (NV16, P210 and P216 surface formats).
+* **Extended output formats:** Decode to various output formats including NV12, YUV420, YUV444, NV16, P010, P016 and RGB24(interleaved and planar)
+
+### **Encode Features**
+
+* **Encoder reconfiguration:** Supports encoder reconfiguration, enabling dynamic updating of encoding parameters without recreating encoder instances.
+* **SEI insertion:** Allows insertion of SEI messages during encoding.
+* **GIL handling:** Improved multithreaded performance through better handling of Global Interpreter Lock (GIL) in C++ layer.
+* **Multi-GPU encode:** Enables multi-GPU encoding to efficiently handle larger workloads.
+* **Codec support:** Support encoding to codec H.264, HEVC, and AV1.
+* **4:2:2 encode:** Supports 4:2:2 encoding for both H.264 and HEVC formats on Blackwell GPUs (NV16 and P210 surface formats).
+* **Extended input formats:** Encode from various input formats including NV12, YV12, IYUV, YUV444, YUV420_10BIT, YUV444_10BIT, NV16, P210, ARGB, ABGR, ARGB10, and ABGR10.
+
+### **Transcode Features**
+
+* **Segment-based transcode:** Enables transcoding of video segments based on timestamp ranges, ideal for content editing and partial processing.
+
+## **What's New**
+
+### **v 2.2.1 (Latest)**
+
+* **Windows on ARM64 support (NVIDIA RTX Spark):** Brings native, hardware-accelerated Python video APIs to Windows on ARM64, enabling high-performance video and AI pipelines on NVIDIA RTX Spark.
+
+### **v 2.2**
+
+* **NVIDIA Jetson support:** Brings easy-to-use, GPU-accelerated Python video APIs to Jetson, making it easy to build video and AI pipelines for robotics and other edge devices.
+* **NVIDIA DGX Spark support:** Unlocks high-throughput, Python-based video decode and encode on DGX Spark, accelerating video curation, experimentation, prototyping, training, and deployment of video and AI pipelines.
+* **Access to in-memory video data:** Reads video directly from Python byte data and streams, allowing video input from network buffers or cloud storage (such as S3) without writing to temporary files, while still supporting random frame access and batch retrieval.
+* **New samples:**
+  * **Face protection:** Jupyter notebook that demonstrates an important privacy use case on Jetson—detecting faces in video with a TensorRT-optimized RetinaFace model and blurring them using GPU-accelerated Gaussian blur to protect identities in video.
+  * **Video decode from memory:** Illustrates decoding video directly from in-memory Python byte data and seekable binary streams (such as bytes, bytearray, memoryview, BytesIO, or file handles).
+  * **Video transcode:** Shows end-to-end GPU video transcoding—demuxing, decoding, encoding, and muxing—using hardware-accelerated video decode and encode.
+  * **Video encode with muxing:** The encode sample now shows how to write encoded video directly into container formats such as MP4 and MOV.
+
+## **Distribution**
+
+PyNvVideoCodec library is distributed in two formats: binary distribution via [PyPI](https://pypi.org/project/pynvvideocodec/) and source code distribution via [NVIDIA NGC](https://catalog.ngc.nvidia.com/orgs/nvidia/resources/pynvvideocodec). In both cases, the library and its dependencies can be installed using a single pip install command.
+
+This package on PyPI contains Python WHLs of the PyNvVideoCodec library and sample applications that demonstrate the use of the PyNvVideoCodec API. To install these please open the shell prompt, and run the following command.
+
+```py
+$ pip install pynvvideocodec
+```
+
+This is the recommended way to install PyNvVideoCodec.
+
+Upon installation of the wheel, the sample applications and documents are placed in the Python site-packages directory. The specific location of site-packages may vary depending on the operating system and Python environment. The path can be identified by running:
+
+```py
+import site; print(site.getsitepackages())
+```
+
+## **Sample Applications and Documents**
+
+* A package containing PyNvVideoCodec source code, Python sample applications and documentation can be downloaded from [NVIDIA NGC](https://catalog.ngc.nvidia.com/orgs/nvidia/resources/pynvvideocodec).
+* For your convenience, the documents are also accessible online at [PyNvVideoCodec Online Documentation](https://docs.nvidia.com/video-technologies/pynvvideocodec/index.html).