saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-05 04:17:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix AdapterSessionTest

Browse source
This commit is contained in:
Jeff Bloomfield 2020-04-25 20:49:51 -07:00
parent f1c19f8495
commit 02e8d10f3a
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
winml/test/adapter/AdapterSessionTest.cpp
View file

@ -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");