@@ -73,17 +73,20 @@ namespace Dml
73
73
bool enableMetacommands,
74
74
bool enableGraphCapture,
75
75
bool enableSyncSpinning,
76
- bool disableMemoryArena) : IExecutionProvider(onnxruntime::kDmlExecutionProvider , OrtDevice(OrtDevice::DML, OrtDevice::MemType::DEFAULT, 0 )) {
77
- D3D12_COMMAND_LIST_TYPE queueType = executionContext->GetCommandListTypeForQueue ();
78
- if (queueType != D3D12_COMMAND_LIST_TYPE_DIRECT && queueType != D3D12_COMMAND_LIST_TYPE_COMPUTE) {
79
- // DML requires either DIRECT or COMPUTE command queues.
80
- ORT_THROW_HR (E_INVALIDARG);
81
- }
76
+ bool disableMemoryArena) :
77
+ IExecutionProvider (onnxruntime::kDmlExecutionProvider , OrtDevice(OrtDevice::DML, OrtDevice::MemType::DEFAULT, 0 ))
78
+ {
79
+ D3D12_COMMAND_LIST_TYPE queueType = executionContext->GetCommandListTypeForQueue ();
80
+ if (queueType != D3D12_COMMAND_LIST_TYPE_DIRECT && queueType != D3D12_COMMAND_LIST_TYPE_COMPUTE)
81
+ {
82
+ // DML requires either DIRECT or COMPUTE command queues.
83
+ ORT_THROW_HR (E_INVALIDARG);
84
+ }
82
85
83
- ComPtr<ID3D12Device> device;
84
- GRAPHICS_THROW_IF_FAILED (dmlDevice->GetParentDevice (IID_GRAPHICS_PPV_ARGS (device.GetAddressOf ())));
86
+ ComPtr<ID3D12Device> device;
87
+ GRAPHICS_THROW_IF_FAILED (dmlDevice->GetParentDevice (IID_GRAPHICS_PPV_ARGS (device.GetAddressOf ())));
85
88
86
- m_impl = wil::MakeOrThrow<ExecutionProviderImpl>(dmlDevice, device.Get (), executionContext, enableMetacommands, enableGraphCapture, enableSyncSpinning, disableMemoryArena);
89
+ m_impl = wil::MakeOrThrow<ExecutionProviderImpl>(dmlDevice, device.Get (), executionContext, enableMetacommands, enableGraphCapture, enableSyncSpinning, disableMemoryArena);
87
90
}
88
91
89
92
std::vector<std::unique_ptr<onnxruntime::ComputeCapability>>
0 commit comments