Start Recording CUDAMemory History(StartRecordCUDAMemoryHistory)

Start Recording CUDAMemory History ノードは、CUDAメモリの割り当てと解放の履歴の追跡記録を開始します。このノードは常に実行されます（バイパスまたはミュートのステータスに関わらず）。これは診断ツールであり、モデル実行時のGPUメモリ使用状況を深く理解し、後のパフォーマンス分析やメモリリークの調査にデータ基盤を提供することを目的としています。このノードは通常、EndRecordCUDAMemoryHistory ノードと対になって使用され、完全な記録サイクルを完了させます。

ノード機能

このノードは、PyTorchの torch.cuda.memory._record_memory_history インターフェースを呼び出すことで中核機能を実現します。ユーザーは、記録の範囲（enabled）、コールスタック情報の詳細度（context と stacks）、および記録エントリの最大数（max_entries）を設定できます。記録を開始する前に、ノードはキャッシュのソフトクリアとピークメモリ統計のリセットを実行し、記録データの正確性を確保します。

ノードパラメータ説明 - Start Recording CUDAMemory History

入力接続 (Inputs)

コントロールパラメータ (Parameters)

出力 (Outputs)

使用シナリオ

複雑な画像生成やモデル推論のワークフローを構築中に、メモリリークが疑われる場合やメモリ使用効率を最適化したい場合に、このノードをワークフローの開始位置に挿入できます。その後、プロセスの終端で EndRecordCUDAMemoryHistory ノードを使用して記録を終了し、スナップショットファイルを保存します。最後に、分析ツール（例: torch.cuda.memory._snapshot()）を使用して詳細なメモリイベント履歴を確認します。

注意事項

このノードは実験的機能であり、その基盤となるAPIはPyTorchのバージョン更新に伴って変更される可能性があることに注意してください。詳細な履歴記録（特に stacks を "all" に設定した場合）を有効にすると、一定のランタイムパフォーマンスオーバーヘッドが生じ、記録情報を保存するために追加のメモリを消費する可能性があります。

Start Recording CUDAMemory History ノードソースコードリンク

Start Recording CUDAMemory History ノードは ComfyUI-KJNodes ノードパッケージに含まれています。