How to fix torch.compile on Windows (※for WaveSpeed)

set PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:32,garbage_collection_threshold:0.8
set TORCH_COMPILE_DEBUG=0
set TORCH_COMPILE_OPT_LEVEL=2
set TORCH_DYNAMO_CONFIG=force_parameter_static_shapes=false
set TRITON_CACHE_DIR=D:\USERFILES\triton (任意のフォルダ指定) 
set TORCH_HOME=D:\USERFILES\torch (任意のフォルダ指定)
set TORCHINDUCTOR_CACHE_DIR=D:\USERFILES\Inductor (任意のフォルダ指定)
set TEMP=D:\USERFILES\Temp (任意のフォルダ指定)
set TEMP=D:\USERFILES\Temp (任意のフォルダ指定)

max_split_size_mb:32      # CUDAメモリの断片化を防ぐための最大分割サイズ（MB）
garbage_collection_threshold:0.8   # メモリの80%使用時にGCを実行

0: 最小限の最適化
1: 標準的な最適化
2: 積極的な最適化（推奨）
3: 最大限の最適化（実験的）

recompiles: 再コンパイルの理由を記録
dynamo: TorchDynamoの動作ログを記録
inductor: Inductorの最適化ログを記録

モデルパラメータの動的な形状変更を許可
再コンパイルの回数を減らす可能性がある

mode = mode.strip() if mode else None

Param(
    [string]$Target = "D:\USERFILES\wan22vace\triton",
    [int]$MaxSizeGB = 50,            # 上限を 50GB に設定
    [int]$KeepMinItems = 2,          # 最低直下アイテム数（フォルダ/ファイルの合計）
    [string[]]$WhitelistPatterns = @("WAN2.2","wan22","wan22vace"), # マッチした名前は削除対象外
    [string]$LogFile = "D:\USERFILES\wan22vace\triton-lru-cleanup-50gb.log",
    [switch]$DryRun                   # DryRun があれば削除は行わない（ログのみ）
)

function Write-Log($msg) {
    $line = "[$(Get-Date -Format 'yyyy-MM-dd HH:mm:ss')] $msg"
    $line | Out-File -FilePath $LogFile -Encoding utf8 -Append
    Write-Output $line
}

Write-Log "=== triton-lru-size-cleanup start (Target=$Target MaxSizeGB=$MaxSizeGB KeepMinItems=$KeepMinItems DryRun=$DryRun) ==="
Write-Log "Whitelist patterns: $($WhitelistPatterns -join ',')"

# 1) 安全確認 - ComfyUI が動いていれば中止
try {
    $comfy = Get-CimInstance Win32_Process -ErrorAction SilentlyContinue |
             Where-Object { ($_.Name -ieq 'python.exe' -or $_.Name -ieq 'python3.exe') -and ($_.CommandLine -match 'ComfyUI') }
} catch {
    $comfy = $null
}

if ($comfy) {
    Write-Log "ComfyUI process detected. Cleanup aborted to avoid race conditions."
    Write-Log "Processes: " + ($comfy | Select-Object ProcessId, CommandLine | Out-String)
    exit 0
}

if (-not (Test-Path $Target)) {
    Write-Log "Target path not found: $Target"
    exit 0
}

# ヘルパー: フォルダ/ファイルのサイズをバイトで取得
function Get-ItemSizeBytes($item) {
    if ($item.PSIsContainer) {
        try {
            $sum = Get-ChildItem -LiteralPath $item.FullName -Recurse -Force -ErrorAction SilentlyContinue |
                   Where-Object { -not $_.PSIsContainer } |
                   Measure-Object -Property Length -Sum
            return [int64]($sum.Sum)
        } catch {
            return 0
        }
    } else {
        return [int64]$item.Length
    }
}

# 現在の合計サイズ（再帰）
function Get-SizeBytes($path) {
    try {
        $sum = Get-ChildItem -LiteralPath $path -Recurse -Force -ErrorAction SilentlyContinue |
               Where-Object { -not $_.PSIsContainer } |
               Measure-Object -Property Length -Sum
        return [int64]$sum.Sum
    } catch {
        return 0
    }
}

$maxBytes = [int64]$MaxSizeGB * 1GB
$currentBytes = Get-SizeBytes -path $Target
Write-Log ("Current size: {0:N0} bytes (~{1:N2} GB)" -f $currentBytes, ($currentBytes / 1GB))

if ($currentBytes -le $maxBytes) {
    Write-Log "Size within limit. No action required."
    Write-Log "=== triton-lru-size-cleanup finished ==="
    exit 0
}

# 取得：triton直下の子要素（ファイルやサブフォルダ）
$children = Get-ChildItem -LiteralPath $Target -Force -ErrorAction SilentlyContinue

if (-not $children) {
    Write-Log "No child items found under $Target"
    exit 0
}

# 各要素に「アクセス時刻キー」とサイズを付与（LastAccessTime が使えない環境では LastWriteTime にフォールバック）
$items = @()
foreach ($c in $children) {
    # ホワイトリスト判定（名前にパターンが含まれていれば除外）
    $isWhitelisted = $false
    foreach ($p in $WhitelistPatterns) {
        if ($c.Name -match [regex]::Escape($p)) {
            $isWhitelisted = $true
            break
        }
    }
    if ($isWhitelisted) {
        Write-Log "Skipping whitelisted item: $($c.FullName)"
        continue
    }

    $accessTime = $null
    try {
        $accessTime = $c.LastAccessTime
        if ($accessTime -eq [datetime]::MinValue -or $accessTime -eq $null) {
            $accessTime = $c.LastWriteTime
        }
    } catch {
        $accessTime = $c.LastWriteTime
    }

    $sizeBytes = Get-ItemSizeBytes -item $c
    $items += [PSCustomObject]@{
        Item = $c
        KeyTime = $accessTime
        SizeBytes = $sizeBytes
    }
}

# 古い順（KeyTime 小さい順）にソート
$items = $items | Sort-Object KeyTime

# 削除ループ（KeepMinItems を守りつつ削る）
foreach ($entry in $items) {
    $remainingChildren = (Get-ChildItem -LiteralPath $Target -Force -ErrorAction SilentlyContinue | Where-Object {
        # 残りカウントでもホワイトリストを除外してカウントする
        $name = $_.Name
        $skip = $false
        foreach ($p in $WhitelistPatterns) { if ($name -match [regex]::Escape($p)) { $skip = $true; break } }
        -not $skip
    }).Count

    if ($remainingChildren -le $KeepMinItems) {
        Write-Log "Reached KeepMinItems ($KeepMinItems). Stop deleting further items."
        break
    }

    if ($currentBytes -le $maxBytes) { break }

    $itm = $entry.Item
    $itemBytes = $entry.SizeBytes
    $timeStr = $entry.KeyTime.ToString('yyyy-MM-dd HH:mm:ss')

    Write-Log ("Candidate: {0} LastAccess/Write: {1} Size: {2:N0} bytes (~{3:N2} GB)" -f $itm.FullName, $timeStr, $itemBytes, ($itemBytes / 1GB))

    if ($DryRun) {
        # DryRun: 実削除しないがサイズを見積もる
        $currentBytes -= $itemBytes
        continue
    }

    try {
        Remove-Item -LiteralPath $itm.FullName -Recurse -Force -ErrorAction Stop
        Write-Log "Removed: $($itm.FullName) freed $itemBytes bytes"
        $currentBytes -= $itemBytes
        Write-Log ("New size estimate: {0:N0} bytes (~{1:N2} GB)" -f $currentBytes, ($currentBytes / 1GB))
    } catch {
        Write-Log "ERROR removing $($itm.FullName): $($_.Exception.Message)"
    }
}

Write-Log ("Final estimated size: {0:N0} bytes (~{1:N2} GB)" -f $currentBytes, ($currentBytes / 1GB))
Write-Log "=== triton-lru-size-cleanup finished ==="

powershell.exe -ExecutionPolicy Bypass -File 設置フォルダ\triton-lru-size-cleanup-50gb.ps1" -DryRun

powershell.exe -ExecutionPolicy Bypass -File "D:\USERFILES\wan22vace\triton-lru-size-cleanup-50gb.ps1"

[2025-08-20 13:49:45] === triton-lru-size-cleanup start (Target=D:\USERFILES\wan22vace\triton MaxSizeGB=50 KeepMinItems=2 DryRun=False) ===
[2025-08-20 13:49:45] Whitelist patterns: WAN2.2,wan22,wan22vace
[2025-08-20 13:49:46] Current size: 0 bytes (~0.00 GB)
[2025-08-20 13:49:46] Size within limit. No action required.
[2025-08-20 13:49:46] === triton-lru-size-cleanup finished ===
PS D:\USERFILES\wan22vace>

schtasks /Create /SC WEEKLY /D SUN /TN "ComfyUI_Triton_LRU_Cleanup_50GB" /TR "powershell.exe -ExecutionPolicy Bypass -File \"D:\USERFILES\wan22vace\triton-lru-size-cleanup-50gb.ps1\"" /ST 04:30 /RL HIGHEST /F

PS D:\USERFILES\wan22vace> schtasks /Query /TN "ComfyUI_Triton_LRU_Cleanup_50GB" /V /FO LIST

フォルダー\
ホスト名:                                       USSOEWWIN
タスク名:                                       \ComfyUI_Triton_LRU_Cleanup_50GB
次回の実行時刻:                                 2025/08/24 13:53:00
状態:                                           準備完了
ログオン モード:                                対話型のみ
前回の実行時刻:                                 1999/11/30 0:00:00
前回の結果:                                     267011
作成者:                                         ussoewwin\ussoewwin
実行するタスク:                                 powershell.exe -ExecutionPolicy Bypass -File " D:\USERFILES\wan22vace\triton-lru-size-cleanup-50gb.ps1\ /ST 12:30 /RL HIGHEST /F
開始:                                           N/A
コメント:                                       N/A
スケジュールされたタスクの状態:                 有効
アイドル時間:                                   無効
電源管理:                                       バッテリ モードで停止, バッテリで開始しない
ユーザーとして実行:                             ussoewwin
再度スケジュールされない場合はタスクを削除する: 無効
タスクを停止するまでの時間:                     72:00:00
スケジュール:                                   スケジュール データをこの形式で使用することはできません。
スケジュールの種類:                             毎週
開始時刻:                                       13:53:00
開始日:                                         2025/08/20
終了日:                                         N/A
日:                                             SUN
月:                                             1 週ごと
繰り返し: 間隔:                                 無効
繰り返し: 終了時刻:                             無効
繰り返し: 期間:                                 無効
繰り返し: 実行中の場合は停止:                   無効
PS D:\USERFILES\wan22vace>

@echo off
del /q /s "D:\USERFILES\wan22vace\Inductor\*"
del /q /s "D:\USERFILES\wan22vace\Temp\*"
del /q /s "D:\USERFILES\wan22vace\triton\*"