Skip to content

feat(wasi-nn): process-wide graph registry, single-flight loads, in-place model updates - #164

Open
ruslanti wants to merge 1 commit into
mainfrom
sync/wasi-nn-graph-registry
Open

ruslanti wants to merge 1 commit into
mainfrom
sync/wasi-nn-graph-registry

Conversation

@ruslanti

Copy link
Copy Markdown
Collaborator

No description provided.

Copilot AI balanced review requested due to automatic review settings October 12, 2026 11:26
@ruslanti ruslanti self-assigned this Oct 12, 2026
@ruslanti
ruslanti requested a review from qrdl October 12, 2026 11:26

Copilot AI left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

🟡 Changes recommended

Revision detection can miss model updates, and evicted live graphs can later be loaded twice.

2 open findings
What changed in this PR

Introduces shared WASI-NN graph caching, single-flight loading, and model revision detection outside request paths.

Changes:

  • Shares graph registries across engines and stores.
  • Preloads models on blocking executor creation.
  • Adds cache lifecycle, revision, and loading tests.
File Description
src/​main.rs Initializes the shared registry.
src/​context.rs Offloads executor compilation and model preload.
crates/​runtime/​src/​store.rs Adds explicit WASI-NN preloading.
crates/​runtime/​src/​service.rs Propagates the shared registry.
crates/​runtime/​src/​registry.rs Implements graph caching and revision tracking.
crates/​runtime/​src/​lib.rs Exposes and threads registry state.
crates/​runtime/​Cargo.toml Adds test filesystem support.
crates/​http-service/​src/​lib.rs Handles optional WASI-NN contexts.
crates/​http-service/​src/​executor/​http.rs Updates test context integration.
Cargo.lock Records the test dependency.

🧠 Review effort: Balanced


💡 Add a code-review agent skill or configure MCP servers for context-aware, tailored reviews. Learn more in the docs.

Comment thread crates/runtime/src/registry.rs
Comment thread crates/runtime/src/registry.rs
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.

2 participants