Skip to content

feat: make startup timeout configurable - #82

Merged
mkhazraee merged 4 commits into
mainfrom
codex/glm53-hicache-startup-budget
Oct 13, 2026
Merged

mkhazraee merged 4 commits into
mainfrom
codex/glm53-hicache-startup-budget

Conversation

@ziqifan617

@ziqifan617 ziqifan617 commented Oct 11, 2026 •

Copy link
Copy Markdown
Contributor

Make the existing KVCR startup timeout configurable per instance via KVCRConfig.startup_timeout_seconds, allowing longer waits for large memory registrations.

None preserves the 30-second default; explicit values must be positive integers. The option applies to the local prepare() and start() waits only, without changing transfer deadlines or requiring matching values across nodes.

Signed-off-by: Ziqi Fan <ziqif@nvidia.com>
@karya0

karya0 commented Oct 12, 2026

Copy link
Copy Markdown
Collaborator

@ziqifan617 : Have you tried: UCX_REG_NONBLOCK_MEM_TYPES=host and UCX_IB_ODP_PREFETCH=n ? These solve the startup problem by using lazy initialization.

@ziqifan617
ziqifan617 marked this pull request as ready for review October 12, 2026 18:00
Signed-off-by: Ziqi Fan <ziqif@nvidia.com>
@ziqifan617
ziqifan617 force-pushed the codex/glm53-hicache-startup-budget branch from 3d265d1 to 4dbf52f Compare October 12, 2026 18:01
@ziqifan617

ziqifan617 commented Oct 12, 2026 •

Copy link
Copy Markdown
Contributor Author

@ziqifan617 : Have you tried: UCX_REG_NONBLOCK_MEM_TYPES=host and UCX_IB_ODP_PREFETCH=n ? These solve the startup problem by using lazy initialization.

Thanks @karya0 , I haven't. Will give it a shot when I have the hw to try it again.

for your suggestion, codex gives below concers:

  1. ODP isn’t universally available. UCX checks device capabilities and can fall back to blocking registration. These host settings also don’t address CUDA-memory registration. UCX implementation
  2. The timeout controls the whole initialization wait, including agent/backend initialization and metadata capture—not only host registration.

But overall, I had met a few times that the registration is slow somehow so exposing it out for future sake as well.

WDYT?

@ziqifan617 ziqifan617 closed this Oct 12, 2026
@ziqifan617
ziqifan617 deleted the codex/glm53-hicache-startup-budget branch October 12, 2026 23:07
@ziqifan617
ziqifan617 restored the codex/glm53-hicache-startup-budget branch October 12, 2026 23:19
@ziqifan617 ziqifan617 reopened this Oct 12, 2026
Signed-off-by: Ziqi Fan <ziqif@nvidia.com>
Signed-off-by: Ziqi Fan <ziqif@nvidia.com>
@ziqifan617 ziqifan617 changed the title fix(startup): allow per-instance timeout for large memory registration feat: make startup timeout configurable Oct 13, 2026
@mkhazraee
mkhazraee merged commit f6b9387 into main Oct 13, 2026
3 checks passed
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.

3 participants