mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-14 20:48:00 +00:00
62c0d24340
* Remove APIs unavailable in Store in #8349, #8178, #8065 * Add UWP stubs of C runtime functions * Remove UWP incompatible tests from UWP build * Remove incompatible tests from Store * Use UWP stubs in store only * Skip partition check outside of Windows * Remove unused WRL include * Workaround Windows header not including what it uses * Fix precompiled header name clash * Workaround SDK bugs * DXCore workaround in Win7 * Fix warning * Fix more warnings * Bump WinML to target Windows 8 * Fix more warnings * Remove unnecessary workarounds * Remove Desktop only APIs from DML adapter
108 lines
3.3 KiB
C++
108 lines
3.3 KiB
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
#include "adapter/pch.h"
|
|
|
|
#include "winml_adapter_c_api.h"
|
|
#include "winml_adapter_apis.h"
|
|
#include "core/session/ort_apis.h"
|
|
|
|
#include <core/providers/winml/winml_provider_factory.h>
|
|
#include <core/providers/cpu/cpu_provider_factory.h>
|
|
|
|
const OrtApi* GetVersion1Api();
|
|
|
|
namespace winmla = Windows::AI::MachineLearning::Adapter;
|
|
|
|
static constexpr WinmlAdapterApi winml_adapter_api_1 = {
|
|
// Schema override
|
|
&winmla::OverrideSchema,
|
|
|
|
// OrtEnv methods
|
|
&winmla::EnvConfigureCustomLoggerAndProfiler,
|
|
|
|
// OrtModel methods
|
|
&winmla::CreateModelFromPath,
|
|
&winmla::CreateModelFromData,
|
|
&winmla::CloneModel,
|
|
&winmla::ModelGetAuthor,
|
|
&winmla::ModelGetName,
|
|
&winmla::ModelGetDomain,
|
|
&winmla::ModelGetDescription,
|
|
&winmla::ModelGetVersion,
|
|
&winmla::ModelGetInputCount,
|
|
&winmla::ModelGetOutputCount,
|
|
&winmla::ModelGetInputName,
|
|
&winmla::ModelGetOutputName,
|
|
&winmla::ModelGetInputDescription,
|
|
&winmla::ModelGetOutputDescription,
|
|
&winmla::ModelGetInputTypeInfo,
|
|
&winmla::ModelGetOutputTypeInfo,
|
|
&winmla::ModelGetMetadataCount,
|
|
&winmla::ModelGetMetadata,
|
|
&winmla::ModelEnsureNoFloat16,
|
|
&winmla::SaveModel,
|
|
|
|
// OrtSessionOptions methods
|
|
&OrtSessionOptionsAppendExecutionProvider_CPU,
|
|
&winmla::OrtSessionOptionsAppendExecutionProviderEx_DML,
|
|
|
|
// OrtSession methods
|
|
&winmla::CreateSessionWithoutModel,
|
|
&winmla::SessionGetExecutionProvider,
|
|
&winmla::SessionInitialize,
|
|
&winmla::SessionRegisterGraphTransformers,
|
|
&winmla::SessionRegisterCustomRegistry,
|
|
&winmla::SessionLoadAndPurloinModel,
|
|
&winmla::SessionStartProfiling,
|
|
&winmla::SessionEndProfiling,
|
|
&winmla::SessionCopyOneInputAcrossDevices,
|
|
&winmla::SessionGetNumberOfIntraOpThreads,
|
|
&winmla::SessionGetIntraOpThreadSpinning,
|
|
&winmla::SessionGetNamedDimensionsOverrides,
|
|
|
|
// Dml methods (TODO need to figure out how these need to move to session somehow...)
|
|
&winmla::DmlExecutionProviderSetDefaultRoundingMode,
|
|
&winmla::DmlExecutionProviderFlushContext,
|
|
&winmla::DmlExecutionProviderReleaseCompletedReferences,
|
|
&winmla::DmlCreateGPUAllocationFromD3DResource,
|
|
&winmla::DmlFreeGPUAllocation,
|
|
&winmla::DmlGetD3D12ResourceFromAllocation,
|
|
&winmla::DmlCopyTensor,
|
|
|
|
&winmla::GetProviderMemoryInfo,
|
|
&winmla::GetProviderAllocator,
|
|
&winmla::FreeProviderAllocator,
|
|
&winmla::GetValueMemoryInfo,
|
|
|
|
&winmla::ExecutionProviderSync,
|
|
|
|
&winmla::CreateCustomRegistry,
|
|
|
|
&winmla::ValueGetDeviceId,
|
|
&winmla::SessionGetInputRequiredDeviceId,
|
|
|
|
&winmla::CreateTensorTypeInfo,
|
|
&winmla::CreateSequenceTypeInfo,
|
|
&winmla::CreateMapTypeInfo,
|
|
&winmla::CreateModel,
|
|
&winmla::ModelAddInput,
|
|
&winmla::ModelAddConstantInput,
|
|
&winmla::ModelAddOutput,
|
|
&winmla::ModelAddOperator,
|
|
&winmla::ModelGetOpsetVersion,
|
|
&winmla::OperatorGetNumInputs,
|
|
&winmla::OperatorGetInputName,
|
|
&winmla::OperatorGetNumOutputs,
|
|
&winmla::OperatorGetOutputName,
|
|
|
|
// Release
|
|
&winmla::ReleaseModel
|
|
};
|
|
|
|
const WinmlAdapterApi* ORT_API_CALL OrtGetWinMLAdapter(_In_ const OrtApi* ort_api) NO_EXCEPTION {
|
|
if (OrtApis::GetApi(2) == ort_api) {
|
|
return &winml_adapter_api_1;
|
|
}
|
|
|
|
return nullptr;
|
|
}
|