| 25 | class HloModuleLoaderTest : public HloTestBase {}; |
| 26 | |
| 27 | TEST_F(HloModuleLoaderTest, StripsLogHeaders) { |
| 28 | const string& hlo_string = R"( |
| 29 | I0521 12:04:45.883483 1509 service.cc:186] HloModule test_log_stripping |
| 30 | I0521 12:04:45.883483 1509 service.cc:186] |
| 31 | I0521 12:04:45.883483 1509 service.cc:186] ENTRY entry { |
| 32 | I0521 12:04:45.883483 1509 service.cc:186] p0 = f32[4]{0} parameter(0) |
| 33 | I0521 12:04:45.883483 1509 service.cc:186] p1 = f32[4]{0} parameter(1) |
| 34 | I0521 12:04:45.883483 1509 service.cc:186] add = f32[4]{0} add(p0, p1) |
| 35 | I0521 12:04:45.883483 1509 service.cc:186] ROOT rooty = (f32[4]{0}, f32[4]{0}) tuple(p1, add) |
| 36 | I0521 12:04:45.883483 1509 service.cc:186] } |
| 37 | )"; |
| 38 | |
| 39 | TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> hlo_module, |
| 40 | LoadModuleFromData(hlo_string, "txt")); |
| 41 | EXPECT_NE(FindInstruction(hlo_module.get(), "p0"), nullptr); |
| 42 | EXPECT_NE(FindInstruction(hlo_module.get(), "p1"), nullptr); |
| 43 | EXPECT_NE(FindInstruction(hlo_module.get(), "add"), nullptr); |
| 44 | EXPECT_NE(FindInstruction(hlo_module.get(), "rooty"), nullptr); |
| 45 | } |
| 46 | |
| 47 | } // namespace |
| 48 | } // namespace xla |
nothing calls this directly
no test coverage detected