Retrieve Device and Command buffer for DML (microsoft#25533)

### Description
Retrieve Device and Command buffer for DML to inject other gfx command
into the same device and command buffer for efficient GPU pipelining



### Motivation and Context
Retrieve Device and Command buffer for DML to inject other gfx command
into the same device and command buffer for efficient GPU pipelining

Author: hisham-hchowdhu

include/onnxruntime/core/providers/dml/dml_provider_factory.h

@@ -151,6 +151,18 @@ struct OrtDmlApi {
    * (high power, low power, or default) and a device filter (None, GPU, or NPU).
    */
   ORT_API2_STATUS(SessionOptionsAppendExecutionProvider_DML2, _In_ OrtSessionOptions* options, OrtDmlDeviceOptions* device_opts);
+
+  /**
+   * GetDMLDevice
+   * returns the DML device attched to the DML execution provider
+   */
+  ORT_API2_STATUS(GetDMLDevice, _In_ OrtSessionOptions* options, _Out_ IDMLDevice** dmlDevice);
+
+  /**
+   * GetDMLCommandQueue
+   * returns the command queue used by DML
+   */
+  ORT_API2_STATUS(GetDMLCommandQueue, _In_ OrtSessionOptions* options, _Out_ ID3D12CommandQueue** dmlCommandQueue);
 };
 
 #ifdef __cplusplus

onnxruntime/core/providers/dml/DmlExecutionProvider/inc/DmlExecutionProvider.h

@@ -52,4 +52,6 @@ namespace Dml
     void RegisterDmlOperators(IMLOperatorRegistry* registry);
     void RegisterCpuOperatorsAsDml(onnxruntime::KernelRegistry* registry);
 
+    void getDMLDevice(onnxruntime::IExecutionProvider* provider, _COM_Outptr_ IDMLDevice** dmlDevice);
+
 } // namespace Dml

onnxruntime/core/providers/dml/dml_provider_factory.cc

@@ -65,6 +65,9 @@ struct DMLProviderFactory : IExecutionProviderFactory {
 
   void SetMetacommandsEnabled(bool metacommands_enabled);
 
+  IDMLDevice* GetDMLDevice();
+  ID3D12CommandQueue* GetDMLCommandQueue();
+
  private:
   ComPtr<IDMLDevice> dml_device_{};
   ComPtr<ID3D12CommandQueue> cmd_queue_{};
@@ -101,6 +104,14 @@ void DMLProviderFactory::SetMetacommandsEnabled(bool metacommands_enabled) {
   metacommands_enabled_ = metacommands_enabled;
 }
 
+IDMLDevice* DMLProviderFactory::GetDMLDevice() {
+  return dml_device_.Get();
+}
+
+ID3D12CommandQueue* DMLProviderFactory::GetDMLCommandQueue() {
+  return cmd_queue_.Get();
+}
+
 std::shared_ptr<IExecutionProviderFactory> CreateExecutionProviderFactory_DML(const ConfigOptions& config_options,
                                                                               IDMLDevice* dml_device,
                                                                               ID3D12CommandQueue* cmd_queue,
@@ -712,13 +723,49 @@ ORT_API_STATUS_IMPL(GetD3D12ResourceFromAllocation, _In_ OrtAllocator* ort_alloc
   API_IMPL_END
 }
 
+ORT_API_STATUS_IMPL(GetDMLDevice, _In_ OrtSessionOptions* options, _Out_ IDMLDevice** dmlDevice) {
+  API_IMPL_BEGIN
+
+  *dmlDevice = nullptr;
+#ifdef USE_DML 
+  if (options) {
+    for (auto& factory : options->provider_factories) {
+      if (auto dml_provider_factory = static_cast<onnxruntime::DMLProviderFactory*>(factory.get())) {
+        *dmlDevice = dml_provider_factory->GetDMLDevice();
+      }
+    }
+  }
+#endif  // USE_DML
+  return nullptr;
+API_IMPL_END
+}
+
+ORT_API_STATUS_IMPL(GetDMLCommandQueue, _In_ OrtSessionOptions* options, _Out_ ID3D12CommandQueue** dmlCommandQ) {
+  API_IMPL_BEGIN
+
+  *dmlCommandQ = nullptr;
+#ifdef USE_DML 
+  if (options) {
+    for (auto& factory : options->provider_factories) {
+      if (auto dml_provider_factory = static_cast<onnxruntime::DMLProviderFactory*>(factory.get())) {
+        *dmlCommandQ = dml_provider_factory->GetDMLCommandQueue();
+      }
+    }
+  }
+#endif  // USE_DML
+  return nullptr;
+  API_IMPL_END
+}
+
 static constexpr OrtDmlApi ort_dml_api_10_to_x = {
   &OrtSessionOptionsAppendExecutionProvider_DML,
   &OrtSessionOptionsAppendExecutionProviderEx_DML,
   &CreateGPUAllocationFromD3DResource,
   &FreeGPUAllocation,
   &GetD3D12ResourceFromAllocation,
   &OrtSessionOptionsAppendExecutionProvider_DML2,
+  &GetDMLDevice,
+  &GetDMLCommandQueue,
 };
 
 const OrtDmlApi* GetOrtDmlApi(_In_ uint32_t /*version*/) NO_EXCEPTION {