[ET-VK] Return fence after waiting is done. (#10808)

pytorchbot · web-flow · commit 4e5ffa320386 · 2025-05-12T11:46:20.000-07:00
Pull Request resolved: #10787 This change returns a fence to fence pool after it has been waited on. ghstack-source-id: 283339435 @exported-using-ghexport Differential Revision: [D74484825](https://our.internmc.facebook.com/intern/diff/D74484825/)
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(`