From 02e8d10f3a86c6ed54cf9678a989fdbe10192445 Mon Sep 17 00:00:00 2001 From: Jeff Bloomfield Date: Sat, 25 Apr 2020 20:49:51 -0700 Subject: [PATCH] Fix AdapterSessionTest --- winml/test/adapter/AdapterSessionTest.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/winml/test/adapter/AdapterSessionTest.cpp b/winml/test/adapter/AdapterSessionTest.cpp index 3185cca865..0244902893 100644 --- a/winml/test/adapter/AdapterSessionTest.cpp +++ b/winml/test/adapter/AdapterSessionTest.cpp @@ -89,7 +89,7 @@ void AppendExecutionProvider_DML() { const auto device = CreateD3DDevice(); const auto queue = CreateD3DQueue(device.get()); - THROW_IF_NOT_OK_MSG(winml_adapter_api->OrtSessionOptionsAppendExecutionProvider_DML(session_options.get(), device.get(), queue.get()), ort_api); + THROW_IF_NOT_OK_MSG(winml_adapter_api->OrtSessionOptionsAppendExecutionProvider_DML(session_options.get(), device.get(), queue.get(), true), ort_api); } void CreateWithoutModel() { @@ -112,7 +112,7 @@ void GetExecutionProvider_DML() { THROW_IF_NOT_OK_MSG(ort_api->DisableMemPattern(session_options.get()), ort_api); const auto device = CreateD3DDevice(); const auto queue = CreateD3DQueue(device.get()); - THROW_IF_NOT_OK_MSG(winml_adapter_api->OrtSessionOptionsAppendExecutionProvider_DML(session_options.get(), device.get(), queue.get()), ort_api); + THROW_IF_NOT_OK_MSG(winml_adapter_api->OrtSessionOptionsAppendExecutionProvider_DML(session_options.get(), device.get(), queue.get(), true), ort_api); const auto model_path = FileHelpers::GetModulePath() + L"fns-candy.onnx"; auto session = CreateUniqueOrtSession(model_path, session_options); @@ -256,7 +256,7 @@ void CopyInputAcrossDevices_DML() { THROW_IF_NOT_OK_MSG(ort_api->DisableMemPattern(session_options.get()), ort_api); const auto device = CreateD3DDevice(); const auto queue = CreateD3DQueue(device.get()); - THROW_IF_NOT_OK_MSG(winml_adapter_api->OrtSessionOptionsAppendExecutionProvider_DML(session_options.get(), device.get(), queue.get()), ort_api); + THROW_IF_NOT_OK_MSG(winml_adapter_api->OrtSessionOptionsAppendExecutionProvider_DML(session_options.get(), device.get(), queue.get(), true), ort_api); auto session = CreateUniqueOrtSession(session_options); LoadAndPurloinModel(session, "fns-candy.onnx");