Fix formatting of DML EP files that was disturbed in an earlier PR. (#…

…22672)
microsoft · Oct 31, 2024 · 69fe58b · 69fe58b
1 parent a2070bf
commit 69fe58b
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 16 deletions.
diff --git a/onnxruntime/core/providers/dml/DmlExecutionProvider/src/DmlExternalBufferAllocator.h b/onnxruntime/core/providers/dml/DmlExecutionProvider/src/DmlExternalBufferAllocator.h
@@ -20,13 +20,15 @@ namespace Dml
     class DmlExternalBufferAllocator : public onnxruntime::IAllocator
     {
     public:
-     DmlExternalBufferAllocator(int device_id) : onnxruntime::IAllocator(
-                                                     OrtMemoryInfo(
-                                                         "DML",
-                                                         OrtAllocatorType::OrtDeviceAllocator,
-                                                         OrtDevice(OrtDevice::DML, OrtDevice::MemType::DEFAULT, 0))) {
-       m_device = onnxruntime::DMLProviderFactoryCreator::CreateD3D12Device(device_id, false);
-     }
+        DmlExternalBufferAllocator(int device_id) : onnxruntime::IAllocator(
+            OrtMemoryInfo(
+                "DML",
+                OrtAllocatorType::OrtDeviceAllocator,
+                OrtDevice(OrtDevice::DML, OrtDevice::MemType::DEFAULT, 0)
+            ))
+        {
+            m_device = onnxruntime::DMLProviderFactoryCreator::CreateD3D12Device(device_id, false);
+        }
 
         void* Alloc(size_t size) final
         {

diff --git a/onnxruntime/core/providers/dml/DmlExecutionProvider/src/ExecutionProvider.cpp b/onnxruntime/core/providers/dml/DmlExecutionProvider/src/ExecutionProvider.cpp
@@ -73,17 +73,20 @@ namespace Dml
         bool enableMetacommands,
         bool enableGraphCapture,
         bool enableSyncSpinning,
-        bool disableMemoryArena) : IExecutionProvider(onnxruntime::kDmlExecutionProvider, OrtDevice(OrtDevice::DML, OrtDevice::MemType::DEFAULT, 0)) {
-      D3D12_COMMAND_LIST_TYPE queueType = executionContext->GetCommandListTypeForQueue();
-      if (queueType != D3D12_COMMAND_LIST_TYPE_DIRECT && queueType != D3D12_COMMAND_LIST_TYPE_COMPUTE) {
-        // DML requires either DIRECT or COMPUTE command queues.
-        ORT_THROW_HR(E_INVALIDARG);
-      }
+        bool disableMemoryArena) :
+            IExecutionProvider(onnxruntime::kDmlExecutionProvider, OrtDevice(OrtDevice::DML, OrtDevice::MemType::DEFAULT, 0))
+    {
+        D3D12_COMMAND_LIST_TYPE queueType = executionContext->GetCommandListTypeForQueue();
+        if (queueType != D3D12_COMMAND_LIST_TYPE_DIRECT && queueType != D3D12_COMMAND_LIST_TYPE_COMPUTE)
+        {
+            // DML requires either DIRECT or COMPUTE command queues.
+            ORT_THROW_HR(E_INVALIDARG);
+        }
 
-      ComPtr<ID3D12Device> device;
-      GRAPHICS_THROW_IF_FAILED(dmlDevice->GetParentDevice(IID_GRAPHICS_PPV_ARGS(device.GetAddressOf())));
+        ComPtr<ID3D12Device> device;
+        GRAPHICS_THROW_IF_FAILED(dmlDevice->GetParentDevice(IID_GRAPHICS_PPV_ARGS(device.GetAddressOf())));
 
-      m_impl = wil::MakeOrThrow<ExecutionProviderImpl>(dmlDevice, device.Get(), executionContext, enableMetacommands, enableGraphCapture, enableSyncSpinning, disableMemoryArena);
+        m_impl = wil::MakeOrThrow<ExecutionProviderImpl>(dmlDevice, device.Get(), executionContext, enableMetacommands, enableGraphCapture, enableSyncSpinning, disableMemoryArena);
     }
 
     std::vector<std::unique_ptr<onnxruntime::ComputeCapability>>