mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-26 22:35:43 +00:00
9169c95a0e
* Support test parameters through CLI arguments * Add WinML do Windows x86/ARM CI builds * Code style fixes * Update googletest Remove GPUTEST macros everywhere now that GTEST_SKIP is supported * Refactor main.cpp * Build scenario tests without DML
70 lines
2.6 KiB
C++
70 lines
2.6 KiB
C++
//-----------------------------------------------------------------------------
|
|
//
|
|
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
|
|
#pragma once
|
|
|
|
// stl
|
|
#include <algorithm>
|
|
#include <codecvt>
|
|
#include <fcntl.h>
|
|
#include <future>
|
|
#include <io.h>
|
|
#include <locale>
|
|
#include <numeric>
|
|
#include <random>
|
|
#include <string_view>
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
// IUnknown must be declared before winrt/base.h is included to light up support for native COM
|
|
// interfaces with C++/WinRT types (e.g. winrt::com_ptr<ITensorNative>).
|
|
#include <Unknwn.h>
|
|
#include "winrt/base.h"
|
|
#include "winrt/Windows.Foundation.Collections.h"
|
|
#include "comp_generated/winrt/windows.ai.machinelearning.h"
|
|
|
|
// WinML
|
|
#include "Windows.AI.MachineLearning.Native.h"
|
|
|
|
#include "runtimeParameters.h"
|
|
|
|
#define EXPECT_THROW_SPECIFIC(statement, exception, condition) \
|
|
EXPECT_THROW( \
|
|
try { \
|
|
statement; \
|
|
} catch (const exception& e) { \
|
|
EXPECT_TRUE(condition(e)); \
|
|
throw; \
|
|
} \
|
|
, exception);
|
|
|
|
// For old versions of gtest without GTEST_SKIP, stream the message and return success instead
|
|
#ifndef GTEST_SKIP
|
|
#define GTEST_SKIP_(message) \
|
|
return GTEST_MESSAGE_(message, ::testing::TestPartResult::kSuccess)
|
|
#define GTEST_SKIP GTEST_SKIP_("")
|
|
#endif
|
|
|
|
#ifndef INSTANTIATE_TEST_SUITE_P
|
|
// Use the old name, removed in newer versions of googletest
|
|
#define INSTANTIATE_TEST_SUITE_P INSTANTIATE_TEST_CASE_P
|
|
#endif
|
|
|
|
#define GPUTEST \
|
|
if (auto noGpuTests = RuntimeParameters::Parameters.find("noGPUtests"); \
|
|
noGpuTests != RuntimeParameters::Parameters.end() && noGpuTests->second != "0") \
|
|
{ \
|
|
GTEST_SKIP() << "GPU tests disabled"; \
|
|
}
|
|
|
|
#define SKIP_EDGECORE \
|
|
if (auto isEdgeCore = RuntimeParameters::Parameters.find("EdgeCore"); \
|
|
isEdgeCore != RuntimeParameters::Parameters.end() && isEdgeCore->second != "0") \
|
|
{ \
|
|
GTEST_SKIP() << "Test can't be run in EdgeCore"; \
|
|
}
|