onnxruntime/winml/test/scenario/cppwinrt/CustomOps.h
Ryan Lai 8d017f5701
Scneario Test : Build Google Test and Taef Test based on preprocessor definition (#2809)
* Add winml macro wrappers on top of google test macros

* change test methods to disabled

* Add custom winml macros for both taef and google tests

* PR comments
2020-01-13 09:22:54 -08:00

19 lines
No EOL
701 B
C

#include "test.h"
struct CustomOpsTestApi
{
SetupTest CustomOpsScenarioTestSetup;
SetupTest CustomOpsScenarioGpuTestSetup;
VoidTest CustomOperatorFusion;
VoidTest CustomKernelWithBuiltInSchema;
VoidTest CustomKernelWithCustomSchema;
};
const CustomOpsTestApi& getapi();
WINML_TEST_CLASS_BEGIN_WITH_SETUP(CustomOpsScenarioTest, CustomOpsScenarioTestSetup)
WINML_TEST(CustomOpsScenarioTest, CustomKernelWithBuiltInSchema)
WINML_TEST(CustomOpsScenarioTest, CustomKernelWithCustomSchema)
WINML_TEST_CLASS_END()
WINML_TEST_CLASS_BEGIN_WITH_SETUP(CustomOpsScenarioGpuTest, CustomOpsScenarioGpuTestSetup)
WINML_TEST(CustomOpsScenarioGpuTest, CustomOperatorFusion)
WINML_TEST_CLASS_END()