Skip to content

fix: move prewarm_migration_speed to after restore (#78) - #79

Merged
MRyutaro merged 2 commits into
mainfrom
fix/prewarm-after-restore
Jun 22, 2026
Merged

MRyutaro merged 2 commits into
mainfrom
fix/prewarm-after-restore

Conversation

@MRyutaro

Copy link
Copy Markdown
Owner

Summary

  • prewarm_migration_speed の呼び出しを ElasticKernel.__init__(起動直後)から ElasticNotebook.load_checkpoint() 完了後に移動
  • 起動時の prewarm NFS I/O が explicit checkpoint の open() と競合し ~0.2s のレイテンシ増加を引き起こす問題を解消 (issue prewarm_migration_speed の NFS I/O が外部の checkpoint と競合する #78)
  • restore 完了後は NFS I/O がアイドル状態のため、競合なし・定常状態での正確な計測が可能

Changes

  • elastic_kernel/kernel.py: __init__ から prewarm_migration_speed() 呼び出しを削除
  • elastic_notebook/elastic_notebook.py: load_checkpoint() 末尾で prewarm_migration_speed() をバックグラウンド実行

Test plan

  • uv run pytest tests/ -x -q — 108 passed, 5 skipped

Closes #78

🤖 Generated with Claude Code

MRyutaro and others added 2 commits June 22, 2026 20:52
#78)

起動時の prewarm NFS I/O が explicit checkpoint と競合し open() レイテンシが
~0.2s に跳ね上がる問題を修正。prewarm を load_checkpoint() 完了後に移動する
ことで、NFS I/O 競合を排除し定常状態での計測を保証する。

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
…FF 時も対応

load_checkpoint() の責務を復元のみに戻し、prewarm_migration_speed() の呼び出しを
kernel.py __init__ の restore 処理直後に配置。restore OFF 時は起動直後に計測する
ため、どちらの構成でも checkpoint クリティカルパス外で計測が完了する。

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
@MRyutaro
MRyutaro merged commit e185b3c into main Jun 22, 2026
7 checks passed
@MRyutaro
MRyutaro deleted the fix/prewarm-after-restore branch June 22, 2026 21:03
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

prewarm_migration_speed の NFS I/O が外部の checkpoint と競合する

1 participant