From 2efab54f9c7ee389dfa9a5f494b043d79290afd6 Mon Sep 17 00:00:00 2001 From: Sheil Kumar Date: Thu, 12 Oct 2023 09:41:45 -0700 Subject: [PATCH] Fix missing registration for new dml device selection API (#17898) Fix missing registration for new dml device selection API Co-authored-by: Sheil Kumar --- onnxruntime/core/providers/dml/dml_provider_factory.cc | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/onnxruntime/core/providers/dml/dml_provider_factory.cc b/onnxruntime/core/providers/dml/dml_provider_factory.cc index df94326e88..cd8bc8fe90 100644 --- a/onnxruntime/core/providers/dml/dml_provider_factory.cc +++ b/onnxruntime/core/providers/dml/dml_provider_factory.cc @@ -514,7 +514,7 @@ ORT_API_STATUS_IMPL(FreeGPUAllocation, _In_ void* ptr) { API_IMPL_END } -ORT_API_STATUS_IMPL(SessionOptionsAppendExecutionProvider_DML2, _In_ OrtSessionOptions* options, OrtDmlDeviceOptions* device_options) { +ORT_API_STATUS_IMPL(OrtSessionOptionsAppendExecutionProvider_DML2, _In_ OrtSessionOptions* options, OrtDmlDeviceOptions* device_options) { API_IMPL_BEGIN #ifdef USE_DML auto factory = onnxruntime::DMLProviderFactoryCreator::CreateFromOptions(device_options); @@ -547,7 +547,8 @@ static constexpr OrtDmlApi ort_dml_api_10_to_x = { &OrtSessionOptionsAppendExecutionProviderEx_DML, &CreateGPUAllocationFromD3DResource, &FreeGPUAllocation, - &GetD3D12ResourceFromAllocation + &GetD3D12ResourceFromAllocation, + &OrtSessionOptionsAppendExecutionProvider_DML2, }; const OrtDmlApi* GetOrtDmlApi(_In_ uint32_t /*version*/) NO_EXCEPTION {