feat: add d4r inference engines and Windows parity tooling

Add native K/M/RR inference engines, HIP kernels, replay and validation harnesses, and Windows DLSS setup and runtime support. Include GPU profiling tools, build and packaging scripts, platform integration patches, and implementation documentation.
2026-10-16 08:54:47 -04:00
1 parent 25493b3198
commit 7c5d3c9157
@@ -0,0 +1,52 @@
+param(
+    [int]$Iterations = 32,
+    [switch]$NoHooks,
+    [switch]$SmallSeed,
+    [switch]$DebugLayer,
+    [switch]$Warp,
+    [switch]$NoHip,
+    [ValidateSet('basic','indirect','indirect-root-reset','initial-pso')][string]$Mode = 'indirect',
+    [string]$OutputDirectory = 'test-results/d3d12-command-development'
+)
+$ErrorActionPreference = 'Stop'
+$repo = (Resolve-Path "$PSScriptRoot/../..").Path
+Push-Location $repo
+$names = @('D4R_COMMAND_PROBE_NO_HOOKS','D4R_COMMAND_PROBE_SMALL_SEED','D4R_COMMAND_DEBUG_SDK','D4R_DIAG_DIR','D4R_COMMAND_PROBE_WARP','D4R_COMMAND_PROBE_NO_HIP')
+$saved = @{}
+foreach ($name in $names) { $saved[$name] = [Environment]::GetEnvironmentVariable($name,'Process') }
+try {
+    if ($Iterations -lt 1 -or $Iterations -gt 1024) { throw 'Iterations must be 1..1024' }
+    $cmake = "$repo/.tools/python/cmake/data/bin/cmake.exe"
+    $build = "$repo/build/windows-rdna4-therock"
+    $hip = "$repo/.tools/therock-10.2.0a20260929/_rocm_sdk_core"
+    $output = [IO.Path]::GetFullPath((Join-Path $repo $OutputDirectory))
+    New-Item -ItemType Directory -Force -Path $output | Out-Null
+    $target = 'd4r_d3d12_command_probe'
+    if ($DebugLayer) {
+        $sdk = "$repo/.tools/agility-1.619.5"
+        if (!(Test-Path "$sdk/build/native/bin/x64/d3d12SDKLayers.dll")) { throw 'Unpack Microsoft.Direct3D.D3D12 1.619.5 into .tools/agility-1.619.5 for this optional debug probe' }
+        & $cmake -S . -B $build "-DD4R_AGILITY_ROOT=$sdk" > "$output/configure.log"
+        if ($LASTEXITCODE) { throw 'Debug probe configuration failed' }
+        $target = 'd4r_d3d12_command_debug_probe'
+    }
+    & $cmake --build $build --target d4r_nvngx_windows $target > "$output/build.log"
+    if ($LASTEXITCODE) { throw "Build failed: $output/build.log" }
+    [Environment]::SetEnvironmentVariable($names[0], $(if ($NoHooks) { '1' } else { $null }), 'Process')
+    [Environment]::SetEnvironmentVariable($names[1], $(if ($SmallSeed) { '1' } else { $null }), 'Process')
+    [Environment]::SetEnvironmentVariable($names[2], $(if ($DebugLayer) { '1' } else { $null }), 'Process')
+    [Environment]::SetEnvironmentVariable($names[3], $output, 'Process')
+    [Environment]::SetEnvironmentVariable($names[4], $(if ($Warp) { '1' } else { $null }), 'Process')
+    [Environment]::SetEnvironmentVariable($names[5], $(if ($NoHip -or $Warp) { '1' } else { $null }), 'Process')
+    # PowerShell 5 treats native stderr as ErrorRecord: retain it without
+    # converting it into a terminating exception, then check the process code.
+    $ErrorActionPreference = 'Continue'
+    & "$build/$target.exe" --hip-root $hip --module "$build/d4r_nvngx.dll" --interop-mode $Mode --iterations $Iterations > "$output/stdout.log" 2> "$output/stderr.log"
+    $code = $LASTEXITCODE
+    $ErrorActionPreference = 'Stop'
+    Get-Content "$output/stdout.log", "$output/stderr.log" | Select-String 'D3D12_VALIDATION|DRED|DEVICE_REMOVED|PROBE_GPU|DIFFERENT|FAIL|PASS|installed'
+    Write-Host "D3D12 command diagnostic logs: $output"
+    if ($code) { throw "D3D12 probe exit=$code" }
+} finally {
+    foreach ($name in $names) { [Environment]::SetEnvironmentVariable($name,$saved[$name],'Process') }
+    Pop-Location
+}