diff --git a/crates/wasi-nn/src/impl.rs b/crates/wasi-nn/src/impl.rs index a8828c1fe9..fb8e781dd1 100644 --- a/crates/wasi-nn/src/impl.rs +++ b/crates/wasi-nn/src/impl.rs @@ -158,7 +158,7 @@ impl<'a> WasiEphemeralNn for WasiNnCtx { }; // Copy the tensor data over to the `out_buffer`. - let mut out_slice = out_buffer.as_array(out_buffer_max_size).as_slice()?; + let mut out_slice = out_buffer.as_array(out_buffer_max_size).as_slice_mut()?; (&mut out_slice[..blob_size as usize]).copy_from_slice(blob.buffer()?); Ok(blob_size)