search

bytecodealliance / wasi-nn

High-level bindings for wasi-nn system calls
Apache License 2.0
45 stars 35 forks source link

Add PyTorch backend for `wasi-nn` crate #62

Closed gusye1234 closed 2 years ago

gusye1234 commented 2 years ago

Hi, I'm trying to implement a PyTorch backend for wasi-nn, and want to write a demo in rust. I hope I can use the wasi-nn crate in my project, but it seems don't have a backend flag for PyTorch.

Should we add one? maybe one more line here: https://github.com/bytecodealliance/wasi-nn/blob/bc63931e0dac2a63533b802980263aee291ed6a4/rust/src/generated.rs#L41 pub const GRAPH_ENCODING_TORCH: GraphEncoding = 1;

abrown commented 2 years ago

Makes sense to me; you may also want to submit a similar change up in the spec repository as well.

abrown commented 2 years ago

This should be resolved by #65.