Skip to content

wasi-nn: Update WIT #11041

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 6 commits into from
Jun 27, 2025
Merged

Conversation

rahulchaphalkar
Copy link
Contributor

@rahulchaphalkar rahulchaphalkar commented Jun 13, 2025

  • Update WIT to latest
  • Updated WIT removes set-input, get-output functions, modifies compute to accept Vec of NamedTensor parameter representing input tensors, and return Vec of NamedTensor representing output tensor.
  • Due to divergence from WITX, compute handles both wit and witx depending on if input tensors in form of Vec is provided.
  • set-input, get-output functions remain in the code for WITX support.
  • Update all backends
  • Update WIT tests, onnx example

@rahulchaphalkar rahulchaphalkar requested review from a team as code owners June 13, 2025 22:02
@rahulchaphalkar rahulchaphalkar requested review from fitzgen and removed request for a team June 13, 2025 22:02
@alexcrichton alexcrichton requested review from abrown and removed request for a team and fitzgen June 13, 2025 22:05
@rahulchaphalkar rahulchaphalkar requested a review from a team as a code owner June 13, 2025 22:26
Copy link
Member

@abrown abrown left a comment

Choose a reason for hiding this comment

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

@rahulchaphalkar walked me through running all the wasi-nn tests with all of the various backends and this PR does not appear to change any results. Adapting this implementation to WITX is getting even more awkward; I really think we should consider removing the WITX half of this crate at some point soon.

@abrown abrown added this pull request to the merge queue Jun 27, 2025
Merged via the queue into bytecodealliance:main with commit a7e1150 Jun 27, 2025
160 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.

2 participants