diff options
Diffstat (limited to 'tools/cnn_test.cc')
| -rw-r--r-- | tools/cnn_test.cc | 3 |
1 files changed, 3 insertions, 0 deletions
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]); |