Skip to content

Commit f06164e

Browse files
authored
[js/web] transfer input buffer back to caller thread (#19677)
### Description When using proxy worker, input buffers should be transferred back to the caller thread after `run()` call is done. Fixes #19488
1 parent 22176a5 commit f06164e

File tree

1 file changed

+1
-1
lines changed
  • js/web/lib/wasm/proxy-worker

1 file changed

+1
-1
lines changed

js/web/lib/wasm/proxy-worker/main.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,7 @@ self.onmessage = (ev: MessageEvent<OrtWasmMessage>): void => {
103103
} else {
104104
postMessage(
105105
{type, out: outputs} as OrtWasmMessage,
106-
extractTransferableBuffers(outputs as SerializableTensorMetadata[]));
106+
extractTransferableBuffers([...inputs, ...outputs] as SerializableTensorMetadata[]));
107107
}
108108
},
109109
err => {

0 commit comments

Comments
 (0)