|
- $ErrorActionPreference = 'Stop'
-
- $vcvars = 'C:\Program Files (x86)\Microsoft Visual Studio\2019\BuildTools\VC\Auxiliary\Build\vcvars64.bat'
- $cudaRoot = 'C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v13.2'
- $nvcc = Join-Path $cudaRoot 'bin\nvcc.exe'
- $src = Join-Path $PSScriptRoot 'internal\demod\gpudemod\native\exports.cu'
- $outDir = Join-Path $PSScriptRoot 'internal\demod\gpudemod\build'
- $dll = Join-Path $outDir 'gpudemod_kernels.dll'
- $lib = Join-Path $outDir 'gpudemod_kernels.lib'
- $exp = Join-Path $outDir 'gpudemod_kernels.exp'
-
- if (!(Test-Path $vcvars)) { throw "vcvars64.bat not found at $vcvars" }
- if (!(Test-Path $nvcc)) { throw "nvcc.exe not found at $nvcc" }
- if (!(Test-Path $src)) { throw "CUDA source not found at $src" }
- if (!(Test-Path $outDir)) { New-Item -ItemType Directory -Path $outDir | Out-Null }
-
- Remove-Item $dll,$lib,$exp -Force -ErrorAction SilentlyContinue
-
- $cmd = @"
- call "$vcvars" && "$nvcc" -shared "$src" -o "$dll" -cudart=hybrid -Xcompiler "/MD" -arch=sm_75 -gencode arch=compute_75,code=sm_75 -gencode arch=compute_80,code=sm_80 -gencode arch=compute_86,code=sm_86 -gencode arch=compute_89,code=sm_89 -gencode arch=compute_90,code=sm_90
- "@
-
- Write-Host 'Building gpudemod CUDA DLL...' -ForegroundColor Cyan
- cmd.exe /c $cmd
- if ($LASTEXITCODE -ne 0) { throw 'gpudemod DLL build failed' }
-
- Write-Host "Built: $dll" -ForegroundColor Green
|
