Skip to content

Commit 136500e

Browse files
authored
fix a memory leak in wasi-nn (#11704)
* fix memory leak * fix clippy::redundant_field_names * fix format
1 parent 5305999 commit 136500e

File tree

1 file changed

+3
-6
lines changed

1 file changed

+3
-6
lines changed

crates/wasi-nn/src/wit.rs

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -249,12 +249,9 @@ impl generated::inference::HostGraphExecutionContext for WasiNnView<'_> {
249249
tracing::debug!("compute with {} inputs", inputs.len());
250250

251251
let mut named_tensors = Vec::new();
252-
for (name, tensor_resopurce) in inputs.iter() {
253-
let tensor = self.table.get(&tensor_resopurce)?;
254-
named_tensors.push(crate::backend::NamedTensor {
255-
name: name.clone(),
256-
tensor: tensor.clone(),
257-
});
252+
for (name, tensor_resopurce) in inputs.into_iter() {
253+
let tensor = self.table.delete(tensor_resopurce)?;
254+
named_tensors.push(crate::backend::NamedTensor { name, tensor });
258255
}
259256

260257
let exec_context = &mut self.table.get_mut(&exec_context)?;

0 commit comments

Comments
 (0)