diff --git a/test_llama_cpp.ps1 b/test_llama_cpp.ps1 deleted file mode 100644 index b98177eac6..0000000000 --- a/test_llama_cpp.ps1 +++ /dev/null @@ -1,234 +0,0 @@ -<# -.SYNOPSIS - Test script for llama.cpp compilation and binary validation on Windows. - Verifies that llama-server was built with CUDA support and can start. - -.USAGE - .\test_llama_cpp.ps1 - .\test_llama_cpp.ps1 -BinaryPath "C:\path\to\llama-server.exe" -#> -param( - [string]$BinaryPath = "" -) - -$ErrorActionPreference = "Continue" - -Write-Host "" -Write-Host "============================================" -ForegroundColor Cyan -Write-Host " llama.cpp Windows Build Test" -ForegroundColor Cyan -Write-Host "============================================" -ForegroundColor Cyan -Write-Host "" - -# -- Step 1: Locate the binary ------------------------------------------ -Write-Host "1. Locating llama-server binary..." -ForegroundColor Yellow - -$SearchPaths = @() - -if ($BinaryPath) { - $SearchPaths += $BinaryPath -} - -# Add all known locations -$RepoRoot = $PSScriptRoot -$SearchPaths += Join-Path $RepoRoot "llama.cpp\build\bin\Release\llama-server.exe" -$SearchPaths += Join-Path $RepoRoot "llama.cpp\build\bin\llama-server.exe" -# Legacy: older setup.ps1 built under ~/.unsloth -$SearchPaths += Join-Path $env:USERPROFILE ".unsloth\llama.cpp\build\bin\Release\llama-server.exe" - -# Check LLAMA_SERVER_PATH env var -$envPath = $env:LLAMA_SERVER_PATH -if ($envPath) { - $SearchPaths = @($envPath) + $SearchPaths -} - -# Also check system PATH -$systemPath = (Get-Command llama-server -ErrorAction SilentlyContinue) -if ($systemPath) { - $SearchPaths += $systemPath.Source -} - -$FoundBinary = $null -foreach ($p in $SearchPaths) { - if (Test-Path $p) { - $FoundBinary = $p - break - } -} - -if (-not $FoundBinary) { - Write-Host " [FAIL] llama-server.exe not found!" -ForegroundColor Red - Write-Host "" - Write-Host " Searched locations:" -ForegroundColor Gray - foreach ($p in $SearchPaths) { - Write-Host " - $p" -ForegroundColor Gray - } - Write-Host "" - Write-Host " To fix: Run setup.bat to build llama.cpp, or set:" -ForegroundColor Yellow - Write-Host ' $env:LLAMA_SERVER_PATH = "C:\path\to\llama-server.exe"' -ForegroundColor Yellow - exit 1 -} - -Write-Host " [OK] Found: $FoundBinary" -ForegroundColor Green - -# -- Step 2: Check file info -------------------------------------------- -Write-Host "" -Write-Host "2. Binary info..." -ForegroundColor Yellow - -$fileInfo = Get-Item $FoundBinary -$sizeMB = [math]::Round($fileInfo.Length / 1MB, 1) -Write-Host " Size: $sizeMB MB" -ForegroundColor Gray -Write-Host " Modified: $($fileInfo.LastWriteTime)" -ForegroundColor Gray - -# -- Step 3: Check for CUDA symbols ------------------------------------ -Write-Host "" -Write-Host "3. Checking for CUDA support..." -ForegroundColor Yellow - -# Run with --help or -v and capture output to check for CUDA indicators -$helpOutput = & $FoundBinary --version 2>&1 | Out-String -if (-not $helpOutput) { - $helpOutput = "" -} - -# Check binary dependencies for CUDA DLLs using dumpbin if available -$dumpbin = (Get-Command dumpbin -ErrorAction SilentlyContinue) -$hasCudaDlls = $false - -if ($dumpbin) { - $deps = & dumpbin /dependents $FoundBinary 2>&1 | Out-String - if ($deps -match "cudart|cublas|cublasLt|nvcuda") { - $hasCudaDlls = $true - Write-Host " [OK] CUDA DLLs found in dependencies (dumpbin)" -ForegroundColor Green - # Extract CUDA DLL names - $cudaDlls = ($deps -split "`n") | Where-Object { $_ -match "cuda|cublas|nvcuda" } | ForEach-Object { $_.Trim() } - foreach ($dll in $cudaDlls) { - if ($dll) { Write-Host " - $dll" -ForegroundColor Gray } - } - } else { - Write-Host " [WARN] No CUDA DLLs found in dependencies!" -ForegroundColor Red - Write-Host " This binary was likely compiled WITHOUT -DGGML_CUDA=ON" -ForegroundColor Red - } -} else { - # Fallback: check file size (CUDA builds are typically > 50MB) - if ($sizeMB -gt 40) { - Write-Host " [LIKELY OK] Binary is $sizeMB MB (CUDA builds are typically > 50MB)" -ForegroundColor Green - } else { - Write-Host " [WARN] Binary is only $sizeMB MB (CPU-only builds are typically < 30MB)" -ForegroundColor Yellow - Write-Host " dumpbin not available for detailed check. Install VS Build Tools." -ForegroundColor Gray - } -} - -# -- Step 4: Quick startup test ----------------------------------------- -Write-Host "" -Write-Host "4. Running startup test (will start and immediately stop)..." -ForegroundColor Yellow - -# Start llama-server on a random port with no model -- just check it initializes -$testPort = Get-Random -Minimum 49152 -Maximum 65535 -$proc = $null - -try { - $proc = Start-Process -FilePath $FoundBinary ` - -ArgumentList "--port", $testPort, "--host", "127.0.0.1" ` - -PassThru -NoNewWindow -RedirectStandardError "$env:TEMP\llama_test_stderr.txt" ` - -RedirectStandardOutput "$env:TEMP\llama_test_stdout.txt" - - # Give it 3 seconds to start - Start-Sleep -Seconds 3 - - # Check if it crashed - if ($proc.HasExited) { - $exitCode = $proc.ExitCode - $stderr = "" - if (Test-Path "$env:TEMP\llama_test_stderr.txt") { - $stderr = Get-Content "$env:TEMP\llama_test_stderr.txt" -Raw - } - $stdout = "" - if (Test-Path "$env:TEMP\llama_test_stdout.txt") { - $stdout = Get-Content "$env:TEMP\llama_test_stdout.txt" -Raw - } - - $allOutput = "$stdout`n$stderr" - - if ($allOutput -match "failed to initialize CUDA") { - Write-Host " [FAIL] CUDA initialization failed!" -ForegroundColor Red - Write-Host " The binary was compiled without CUDA support or CUDA drivers are missing." -ForegroundColor Red - Write-Host "" - Write-Host " Rebuild with: cmake -DGGML_CUDA=ON ..." -ForegroundColor Yellow - } elseif ($allOutput -match "HTTPS is not supported") { - # This is expected when LLAMA_CURL=OFF -- not a real failure - Write-Host " [OK] Binary started (HTTPS warning is expected -- we use local files)" -ForegroundColor Green - } else { - Write-Host " [WARN] Process exited with code $exitCode" -ForegroundColor Yellow - } - - if ($allOutput.Trim()) { - Write-Host "" - Write-Host " --- Output ---" -ForegroundColor Gray - $allOutput.Trim().Split("`n") | ForEach-Object { Write-Host " $_" -ForegroundColor Gray } - } - } else { - Write-Host " [OK] llama-server started successfully on port $testPort" -ForegroundColor Green - - # Check CUDA detection from startup output - Start-Sleep -Seconds 1 - $stderr = "" - if (Test-Path "$env:TEMP\llama_test_stderr.txt") { - $stderr = Get-Content "$env:TEMP\llama_test_stderr.txt" -Raw - } - - if ($stderr -match "CUDA") { - if ($stderr -match "failed to initialize CUDA") { - Write-Host " [FAIL] CUDA init failed at runtime!" -ForegroundColor Red - } else { - Write-Host " [OK] CUDA detected at runtime" -ForegroundColor Green - } - } - - # Kill it - Stop-Process -Id $proc.Id -Force -ErrorAction SilentlyContinue - Write-Host " Stopped test server." -ForegroundColor Gray - } -} catch { - Write-Host " [ERROR] Could not start llama-server: $_" -ForegroundColor Red -} finally { - if ($proc -and -not $proc.HasExited) { - Stop-Process -Id $proc.Id -Force -ErrorAction SilentlyContinue - } - Remove-Item "$env:TEMP\llama_test_stderr.txt" -ErrorAction SilentlyContinue - Remove-Item "$env:TEMP\llama_test_stdout.txt" -ErrorAction SilentlyContinue -} - -# -- Step 5: Check llama-quantize --------------------------------------- -Write-Host "" -Write-Host "5. Checking llama-quantize..." -ForegroundColor Yellow - -$quantizePath = Join-Path (Split-Path $FoundBinary) "llama-quantize.exe" -if (Test-Path $quantizePath) { - $qSize = [math]::Round((Get-Item $quantizePath).Length / 1MB, 1) - Write-Host " [OK] Found: $quantizePath ($qSize MB)" -ForegroundColor Green -} else { - Write-Host " [WARN] llama-quantize.exe not found alongside llama-server" -ForegroundColor Yellow - Write-Host " GGUF export/quantization won't work without it" -ForegroundColor Yellow -} - -# -- Summary ------------------------------------------------------------ -Write-Host "" -Write-Host "============================================" -ForegroundColor Cyan -Write-Host " Summary" -ForegroundColor Cyan -Write-Host "============================================" -ForegroundColor Cyan -Write-Host " Binary: $FoundBinary" -ForegroundColor Gray -Write-Host " Size: $sizeMB MB" -ForegroundColor Gray -if ($hasCudaDlls) { - Write-Host " CUDA: YES (confirmed via DLL deps)" -ForegroundColor Green -} elseif ($sizeMB -gt 40) { - Write-Host " CUDA: LIKELY (large binary size)" -ForegroundColor Yellow -} else { - Write-Host " CUDA: NO (rebuild with -DGGML_CUDA=ON)" -ForegroundColor Red -} -Write-Host "" - -if (-not $hasCudaDlls -and $sizeMB -le 40) { - Write-Host "To rebuild with CUDA:" -ForegroundColor Yellow - Write-Host ' 1. Delete the build dir: Remove-Item -Recurse -Force "$env:USERPROFILE\.unsloth\llama.cpp\build"' -ForegroundColor Gray - Write-Host ' 2. Re-run: .\setup.bat' -ForegroundColor Gray - Write-Host "" -}