saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-27 20:02:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Avoid command list reset in common case of re-used command list execution

Browse source
This commit is contained in:
Jeff Bloomfield 2023-09-19 20:34:47 -07:00
parent 0991c4b43d
commit adb2dd3eec
1 changed files with 18 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
onnxruntime/core/providers/dml/DmlExecutionProvider/src/DmlCommandRecorder.cpp
View file

@ -251,6 +251,24 @@ void DmlCommandRecorder::ExecuteCommandList(
_Out_ uint64_t* completionValue
)
{
if (!m_operationsRecordedInCurrentCommandList)
{
// The caller can re-use relevant resources after the next set of work to be
// flushed has completed. Its command list hasn't been executed yet, just batched.
GpuEvent gpuEvent = m_queue->GetNextCompletionEvent();
gpuEvent.fence.CopyTo(fence);
*completionValue = gpuEvent.fenceValue;
m_queue->ExecuteCommandLists(
gsl::span<ID3D12CommandList*>(reinterpret_cast<ID3D12CommandList**>(&commandList), 1));
// Fail early if something horrifying happens
ORT_THROW_IF_FAILED(m_dmlDevice->GetDeviceRemovedReason());
ORT_THROW_IF_FAILED(m_d3dDevice->GetDeviceRemovedReason());
return;
}
ORT_THROW_IF_FAILED(m_currentCommandList->Close());
if (m_operationsRecordedInCurrentCommandList)