[ET-VK] Return fence after waiting is done.

trivedivivek · web-flow · commit 9ebf5efb4a4b · 2025-05-11T20:35:38.000-07:00
Differential Revision: D74484825 Pull Request resolved: #10787
diff --git a/backends/vulkan/runtime/graph/ComputeGraph.cpp b/backends/vulkan/runtime/graph/ComputeGraph.cpp
@@ -625,6 +625,7 @@ void ComputeGraph::prepack() const {
   vkapi::VulkanFence fence = context_->fences().get_fence();
   context_->submit_cmd_to_gpu(fence.get_submit_handle(), /*final_use = */ true);
   fence.wait();
+  context_->fences().return_fence(fence);
 
   context_->flush();
 }
@@ -649,6 +650,7 @@ void ComputeGraph::execute() const {
   vkapi::VulkanFence fence = context_->fences().get_fence();
   context_->submit_cmd_to_gpu(fence.get_submit_handle());
   fence.wait();
+  context_->fences().return_fence(fence);
 }
 
 void ComputeGraph::resize_input(

Original file line number	Diff line number	Diff line change
`@@ -625,6 +625,7 @@ void ComputeGraph::prepack() const {`
`625`	`625`	`vkapi::VulkanFence fence = context_->fences().get_fence();`
`626`	`626`	`context_->submit_cmd_to_gpu(fence.get_submit_handle(), /final_use = / true);`
`627`	`627`	`fence.wait();`
	`628`	`+ context_->fences().return_fence(fence);`
`628`	`629`
`629`	`630`	`context_->flush();`
`630`	`631`	`}`
`@@ -649,6 +650,7 @@ void ComputeGraph::execute() const {`
`649`	`650`	`vkapi::VulkanFence fence = context_->fences().get_fence();`
`650`	`651`	`context_->submit_cmd_to_gpu(fence.get_submit_handle());`
`651`	`652`	`fence.wait();`
	`653`	`+ context_->fences().return_fence(fence);`
`652`	`654`	`}`
`653`	`655`
`654`	`656`	`void ComputeGraph::resize_input(`