| 56 | }; |
| 57 | |
| 58 | TEST_F(TupleSimplifierTest, TupleOfParameters) { |
| 59 | // A Tuple constructed of a bunch of parameters should not be changed. |
| 60 | HloComputation::Builder builder(TestName()); |
| 61 | HloInstruction* param0 = builder.AddInstruction( |
| 62 | HloInstruction::CreateParameter(0, scalar_shape_, "param0")); |
| 63 | HloInstruction* param1 = builder.AddInstruction( |
| 64 | HloInstruction::CreateParameter(1, scalar_shape_, "param1")); |
| 65 | HloInstruction* param2 = builder.AddInstruction( |
| 66 | HloInstruction::CreateParameter(2, scalar_shape_, "param2")); |
| 67 | builder.AddInstruction(HloInstruction::CreateTuple({param0, param1, param2})); |
| 68 | auto module = CreateNewVerifiedModule(); |
| 69 | module->AddEntryComputation(builder.Build()); |
| 70 | |
| 71 | Run(module.get(), /*change_expected=*/false); |
| 72 | } |
| 73 | |
| 74 | TEST_F(TupleSimplifierTest, GteOfTupleOfParameter) { |
| 75 | // A GTE of a tuple parameter should not be changed. |
nothing calls this directly
no test coverage detected