mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-22 22:01:08 +00:00
3787a071c5
* learning model doesn't need lotusEnvironment and CPU shouldn't include dmlEP headers * User/xianz/win ml telemetry (#2410) * add option to enable winml telemetry * add option to enable winml telemetry * clean logs while developping * clean the log of GUID * compile onnxruntime_common with winml telemetry * use option for use_telemetry * rename option winml_use_telemetry to onnxruntime_use_telemetry * little change * Add opset and IR check when loading model (#2413) * Add opset and IR check. * Add test case for future opsets. https://github.com/microsoft/onnxruntime/issues/2371 * WinML CI (#2412) * Pass flags to build/test WinML in CI * Add initial CMake config for unit tests in WinML * Set winml_unittests standard to C++17 * Add WinML API tests and port them to googletest * Install WinML test collateral * Add LearningModelSessionAPITests ported to googletest * Fix WinML test files encoding * Add GPU tests * Add parameterized test, skip GPU tests * Enable precompiled header * Remove unused code and collateral * Remove brand images * Add dllload.cpp * Remove images not used in API tests * Add LICENSE.md to image collaterals * Add models with licenses * Remove FNS Candy tests * Add API test models * Add ModelInSubdirectory * Install collaterals post-build with copy_if_different, split common lib * fix warnings * Link to gtest_main * fix bad merge
41 lines
1.2 KiB
C++
41 lines
1.2 KiB
C++
//-----------------------------------------------------------------------------
|
|
//
|
|
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
|
|
#pragma once
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
class APITest : public ::testing::Test
|
|
{
|
|
protected:
|
|
void LoadModel(const std::wstring& modelPath)
|
|
{
|
|
std::wstring fullPath = FileHelpers::GetModulePath() + modelPath;
|
|
m_model = winrt::Windows::AI::MachineLearning::LearningModel::LoadFromFilePath(fullPath);
|
|
}
|
|
|
|
winrt::Windows::AI::MachineLearning::LearningModel m_model = nullptr;
|
|
winrt::Windows::AI::MachineLearning::LearningModelDevice m_device = nullptr;
|
|
winrt::Windows::AI::MachineLearning::LearningModelSession m_session = nullptr;
|
|
|
|
uint64_t GetAdapterIdQuadPart()
|
|
{
|
|
LARGE_INTEGER id;
|
|
id.LowPart = m_device.AdapterId().LowPart;
|
|
id.HighPart = m_device.AdapterId().HighPart;
|
|
return id.QuadPart;
|
|
};
|
|
|
|
_LUID GetAdapterIdAsLUID()
|
|
{
|
|
_LUID id;
|
|
id.LowPart = m_device.AdapterId().LowPart;
|
|
id.HighPart = m_device.AdapterId().HighPart;
|
|
return id;
|
|
}
|
|
|
|
bool m_runGPUTests = true;
|
|
};
|