mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-18 21:21:17 +00:00
eb7486d632
* Add DML EP tests * wip * Fix DmlCopyTensor * Cleanup * Remove leaks in AdapterDmlEpTest * Fix rebase conflicts * Move GPUTEST macro to tests instead of suite setup * Address PR feedback
42 lines
1.7 KiB
C
42 lines
1.7 KiB
C
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#include "test.h"
|
|
struct AdapterDmlEpTestApi
|
|
{
|
|
SetupTest AdapterDmlEpTestSetup;
|
|
TeardownClass AdapterDmlEpTestTeardown;
|
|
VoidTest DmlExecutionProviderSetDefaultRoundingMode;
|
|
VoidTest DmlExecutionProviderFlushContext;
|
|
VoidTest DmlExecutionProviderReleaseCompletedReferences;
|
|
VoidTest DmlCreateGPUAllocationFromD3DResource;
|
|
VoidTest DmlCreateAndFreeGPUAllocationFromD3DResource;
|
|
VoidTest GetProviderMemoryInfo;
|
|
VoidTest GetAndFreeProviderAllocator;
|
|
VoidTest GetValueMemoryInfo;
|
|
VoidTest ExecutionProviderSync;
|
|
VoidTest DmlCopyTensor;
|
|
VoidTest CreateCustomRegistry;
|
|
VoidTest ValueGetDeviceId;
|
|
VoidTest SessionGetInputRequiredDeviceId;
|
|
};
|
|
const AdapterDmlEpTestApi& getapi();
|
|
|
|
WINML_TEST_CLASS_BEGIN(AdapterDmlEpTest)
|
|
WINML_TEST_CLASS_SETUP_METHOD(AdapterDmlEpTestSetup)
|
|
WINML_TEST_CLASS_TEARDOWN_METHOD(AdapterDmlEpTestTeardown)
|
|
WINML_TEST_CLASS_BEGIN_TESTS
|
|
WINML_TEST(AdapterDmlEpTest, DmlExecutionProviderSetDefaultRoundingMode)
|
|
WINML_TEST(AdapterDmlEpTest, DmlExecutionProviderFlushContext)
|
|
WINML_TEST(AdapterDmlEpTest, DmlExecutionProviderReleaseCompletedReferences)
|
|
WINML_TEST(AdapterDmlEpTest, DmlCreateGPUAllocationFromD3DResource)
|
|
WINML_TEST(AdapterDmlEpTest, DmlCreateAndFreeGPUAllocationFromD3DResource)
|
|
WINML_TEST(AdapterDmlEpTest, GetProviderMemoryInfo)
|
|
WINML_TEST(AdapterDmlEpTest, GetAndFreeProviderAllocator)
|
|
WINML_TEST(AdapterDmlEpTest, GetValueMemoryInfo)
|
|
WINML_TEST(AdapterDmlEpTest, ExecutionProviderSync)
|
|
WINML_TEST(AdapterDmlEpTest, DmlCopyTensor)
|
|
WINML_TEST(AdapterDmlEpTest, CreateCustomRegistry)
|
|
WINML_TEST(AdapterDmlEpTest, ValueGetDeviceId)
|
|
WINML_TEST(AdapterDmlEpTest, SessionGetInputRequiredDeviceId)
|
|
WINML_TEST_CLASS_END()
|