wasi-nn: fix mutable guestslice borrow

This commit is contained in:
Pat Hickey
2020-11-19 15:28:53 -08:00
parent fb68c80420
commit 224e8b0e88

View File

@@ -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)