| 533 | }; |
| 534 | |
| 535 | TEST_P(ConcatR2BinaryTest, DoIt) { |
| 536 | const R2BinarySpec& spec = GetParam(); |
| 537 | Array2D<int32> lhs(spec.lhs_dim0, spec.lhs_dim1); |
| 538 | lhs.FillUnique(); |
| 539 | Array2D<int32> rhs(spec.rhs_dim0, spec.rhs_dim1); |
| 540 | rhs.FillUnique(1000); |
| 541 | |
| 542 | XlaBuilder builder(TestName()); |
| 543 | auto a0 = ConstantR2FromArray2D<int32>(&builder, lhs); |
| 544 | auto a1 = ConstantR2FromArray2D<int32>(&builder, rhs); |
| 545 | ConcatInDim(&builder, {a0, a1}, spec.concat_dimension); |
| 546 | |
| 547 | std::unique_ptr<Array2D<int32>> expected = |
| 548 | ReferenceUtil::Concat2D(lhs, rhs, spec.concat_dimension); |
| 549 | ComputeAndCompareR2<int32>(&builder, *expected, {}); |
| 550 | } |
| 551 | |
| 552 | // Regression test for b/31944287. x*y is used (at the same index) by all |
| 553 | // operands of the concat. We should emit x*y in three incoming basic blocks of |
nothing calls this directly
no test coverage detected