diff options
| -rw-r--r-- | src/gpu/texture_readback.cc | 1 | ||||
| -rw-r--r-- | tools/cnn_test.cc | 3 |
2 files changed, 4 insertions, 0 deletions
diff --git a/src/gpu/texture_readback.cc b/src/gpu/texture_readback.cc index f3e4056..e25da9e 100644 --- a/src/gpu/texture_readback.cc +++ b/src/gpu/texture_readback.cc @@ -71,6 +71,7 @@ std::vector<uint8_t> read_texture_pixels( wgpuQueueSubmit(queue, 1, &commands); wgpuCommandBufferRelease(commands); wgpuCommandEncoderRelease(encoder); + wgpuQueueRelease(queue); // Release the queue reference // Wait for copy to complete before mapping wgpuDevicePoll(device, true, nullptr); diff --git a/tools/cnn_test.cc b/tools/cnn_test.cc index 3c96800..4ce9637 100644 --- a/tools/cnn_test.cc +++ b/tools/cnn_test.cc @@ -525,6 +525,9 @@ int main(int argc, char** argv) { } } + // Wait for all GPU work to complete before cleanup + wgpuDevicePoll(device, true, nullptr); + // Cleanup wgpuTextureViewRelease(intermediate_views[0]); wgpuTextureViewRelease(intermediate_views[1]); |