| 31 | class TestUtilsTest : public LocalClientTestBase {}; |
| 32 | |
| 33 | XLA_TEST_F(TestUtilsTest, UnusedParam) { |
| 34 | XlaBuilder builder(TestName()); |
| 35 | // Make the reduction lambda. |
| 36 | Shape single_float = ShapeUtil::MakeShape(F32, {}); |
| 37 | Parameter(&builder, 0, single_float, "unused"); |
| 38 | Parameter(&builder, 1, single_float, "used"); |
| 39 | auto computation_status = builder.Build(); |
| 40 | TF_ASSERT_OK(computation_status.status()); |
| 41 | |
| 42 | // Make the reduction. |
| 43 | Shape pair_float = ShapeUtil::MakeShape(F32, {2}); |
| 44 | Reduce(Parameter(&builder, 0, pair_float, "operand"), |
| 45 | Parameter(&builder, 1, single_float, "init"), |
| 46 | computation_status.ValueOrDie(), {0}); |
| 47 | computation_status = builder.Build(); |
| 48 | TF_ASSERT_OK(computation_status.status()); |
| 49 | |
| 50 | TF_ASSERT_OK_AND_ASSIGN( |
| 51 | auto executables, local_client_->Compile(computation_status.ValueOrDie(), |
| 52 | {&pair_float, &single_float}, |
| 53 | ExecutableBuildOptions())); |
| 54 | HloModule& module = |
| 55 | const_cast<HloModule&>(executables[0]->executable()->module()); |
| 56 | TF_ASSERT_OK(MakeFakeArguments(&module).status()); |
| 57 | } |
| 58 | |
| 59 | XLA_TEST_F(TestUtilsTest, MultipleIndexSpacesForDynamicSlices) { |
| 60 | auto module = ParseAndReturnVerifiedModule( |
nothing calls this directly
no test coverage detected