Fix build of wasi-nn

This commit is contained in:
Alex Crichton
2021-02-17 14:16:18 -08:00
committed by Andrew Brown
parent fa98f0bc91
commit 250f11c572
5 changed files with 10 additions and 10 deletions

View File

@@ -12,7 +12,7 @@ use wiggle::GuestPtr;
#[derive(Debug, Error)]
pub enum UsageError {
#[error("Only OpenVINO's IR is currently supported, passed encoding: {0}")]
#[error("Only OpenVINO's IR is currently supported, passed encoding: {0:?}")]
InvalidEncoding(GraphEncoding),
#[error("OpenVINO expects only two buffers (i.e. [ir, weights]), passed: {0}")]
InvalidNumberOfBuilders(u32),

View File

@@ -18,7 +18,6 @@ wasmtime_wiggle::wasmtime_integration!({
wasi_ephemeral_nn => {
name: WasiNn,
docs: "An instantiated instance of the wasi-nn exports.",
function_override: {}
}
},
});

View File

@@ -6,23 +6,23 @@ use crate::ctx::WasiNnError;
wiggle::from_witx!({
witx: ["$WASI_ROOT/phases/ephemeral/witx/wasi_ephemeral_nn.witx"],
ctx: WasiNnCtx,
errors: { errno => WasiNnError }
errors: { nn_errno => WasiNnError }
});
use types::Errno;
use types::NnErrno;
/// Wiggle generates code that performs some input validation on the arguments passed in by users of
/// wasi-nn. Here we convert the validation error into one (or more, eventually) of the error
/// variants defined in the witx.
impl types::GuestErrorConversion for WasiNnCtx {
fn into_errno(&self, e: wiggle::GuestError) -> Errno {
fn into_nn_errno(&self, e: wiggle::GuestError) -> NnErrno {
eprintln!("Guest error: {:?}", e);
Errno::InvalidArgument
NnErrno::InvalidArgument
}
}
impl<'a> types::UserErrorConversion for WasiNnCtx {
fn errno_from_wasi_nn_error(&self, e: WasiNnError) -> Result<Errno, wiggle::Trap> {
fn nn_errno_from_wasi_nn_error(&self, e: WasiNnError) -> Result<NnErrno, wiggle::Trap> {
eprintln!("Host error: {:?}", e);
match e {
WasiNnError::OpenvinoError(_) => unimplemented!(),
@@ -33,7 +33,7 @@ impl<'a> types::UserErrorConversion for WasiNnCtx {
}
/// Additionally, we must let Wiggle know which of our error codes represents a successful operation.
impl wiggle::GuestErrorType for Errno {
impl wiggle::GuestErrorType for NnErrno {
fn success() -> Self {
Self::Success
}