FaSTTUBe/GST-Tensordecoder-ov_ep
37
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

d3d12encoder: Handle external fence explicitly

Browse Source 
Waits for external fence if any

Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/6749>
This commit is contained in:
Seungha Yang 2024-04-27 23:30:40 +09:00 committed by GStreamer Marge Bot
parent 2a14793ee1
commit 2c203e0d40
1 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
subprojects/gst-plugins-bad/sys/d3d12/gstd3d12encoder.cpp
View File

@ -736,6 +736,12 @@ gst_d3d12_encoder_upload_frame (GstD3D12Encoder * self, GstBuffer * buffer)
std::vector < GstD3D12CopyTextureRegionArgs > copy_args; std::vector < GstD3D12CopyTextureRegionArgs > copy_args;
auto dmem = GST_D3D12_MEMORY_CAST (mem); auto dmem = GST_D3D12_MEMORY_CAST (mem);
auto src_resource = gst_d3d12_memory_get_resource_handle (dmem); auto src_resource = gst_d3d12_memory_get_resource_handle (dmem);
ComPtr < ID3D12Fence > fence_to_wait;
guint64 fence_val_to_wait = 0;
gst_d3d12_memory_get_external_fence (dmem,
&fence_to_wait, &fence_val_to_wait);
dmem = (GstD3D12Memory *) gst_buffer_peek_memory (upload, 0); dmem = (GstD3D12Memory *) gst_buffer_peek_memory (upload, 0);
auto dst_resource = gst_d3d12_memory_get_resource_handle (dmem); auto dst_resource = gst_d3d12_memory_get_resource_handle (dmem);
D3D12_BOX src_box[2]; D3D12_BOX src_box[2];
@ -771,8 +777,8 @@ gst_d3d12_encoder_upload_frame (GstD3D12Encoder * self, GstBuffer * buffer)
guint64 fence_val = 0; guint64 fence_val = 0;
gst_d3d12_device_copy_texture_region (self->device, copy_args.size (), gst_d3d12_device_copy_texture_region (self->device, copy_args.size (),
copy_args.data (), nullptr, nullptr, 0, D3D12_COMMAND_LIST_TYPE_DIRECT, copy_args.data (), nullptr, fence_to_wait.Get (), fence_val_to_wait,
&fence_val); D3D12_COMMAND_LIST_TYPE_DIRECT, &fence_val);
gst_d3d12_buffer_after_write (upload, fence_val); gst_d3d12_buffer_after_write (upload, fence_val);
} else { } else {
GstVideoFrame src_frame, dst_frame; GstVideoFrame src_frame, dst_frame;
@ -1425,6 +1431,14 @@ gst_d3d12_encoder_handle_frame (GstVideoEncoder * encoder,
return GST_FLOW_ERROR; return GST_FLOW_ERROR;
} }
ComPtr < ID3D12Fence > fence_to_wait;
guint64 fence_val_to_wait = 0;
if (gst_d3d12_memory_get_external_fence (mem,
&fence_to_wait, &fence_val_to_wait)) {
gst_d3d12_command_queue_execute_wait (priv->cmd->queue,
fence_to_wait.Get (), fence_val_to_wait);
}
auto completed = gst_d3d12_device_get_completed_value (self->device, auto completed = gst_d3d12_device_get_completed_value (self->device,
D3D12_COMMAND_LIST_TYPE_DIRECT); D3D12_COMMAND_LIST_TYPE_DIRECT);
if (completed < mem->fence_value) { if (completed < mem->fence_value) {