| 131 | } |
| 132 | |
| 133 | TEST(FunctionApiInfoTest, ParseTags) { |
| 134 | FunctionDefLibrary func_lib; |
| 135 | PopulateSampleLibrary(/* mismatch_args */ false, &func_lib); |
| 136 | FunctionLibraryApiInfo lib_api_info; |
| 137 | TF_ASSERT_OK(lib_api_info.Init(func_lib)); |
| 138 | |
| 139 | EXPECT_EQ("DoStuff", GetInterfaceName(lib_api_info, "DoStuffCpu")); |
| 140 | EXPECT_EQ("DoStuff", GetInterfaceName(lib_api_info, "DoStuffGpu")); |
| 141 | EXPECT_EQ("DoThings", GetInterfaceName(lib_api_info, "DoThings")); |
| 142 | |
| 143 | EXPECT_EQ("CPU", GetPreferredDevice(lib_api_info, "DoStuffCpu")); |
| 144 | EXPECT_EQ("GPU", GetPreferredDevice(lib_api_info, "DoStuffGpu")); |
| 145 | EXPECT_EQ("", GetPreferredDevice(lib_api_info, "DoThings")); |
| 146 | |
| 147 | EXPECT_TRUE(CheckEquivImpl(lib_api_info, "DoStuffCpu", {"DoStuffGpu"})); |
| 148 | EXPECT_TRUE(CheckEquivImpl(lib_api_info, "DoStuffGpu", {"DoStuffCpu"})); |
| 149 | EXPECT_TRUE(CheckEquivImpl(lib_api_info, "Undefined", {})); |
| 150 | EXPECT_TRUE(CheckEquivImpl(lib_api_info, "OneOff", {})); |
| 151 | EXPECT_TRUE(CheckEquivImpl(lib_api_info, "AnotherOneOff", {})); |
| 152 | EXPECT_TRUE(CheckEquivImpl(lib_api_info, "DoThings", {})); |
| 153 | } |
| 154 | |
| 155 | TEST(FunctionApiInfoTest, ComplexFunctionLib) { |
| 156 | FunctionDefLibrary func_lib; |
nothing calls this directly
no test coverage detected