| 580 | testNameGenerator<ReduceByKeyP>); |
| 581 | |
| 582 | TEST_P(ReduceByKeyP, SumDim0) { |
| 583 | if (noHalfTests(GetParam()->vType_)) { |
| 584 | GTEST_SKIP() << "Half not supported on this device"; |
| 585 | } |
| 586 | if (noHalfTests(GetParam()->kType_)) { |
| 587 | GTEST_SKIP() << "Half not supported on this device"; |
| 588 | } |
| 589 | if (noDoubleTests(GetParam()->vType_)) { |
| 590 | GTEST_SKIP() << "Double not supported on this device"; |
| 591 | } |
| 592 | array keyRes, valsReduced; |
| 593 | sumByKey(keyRes, valsReduced, keys, vals, 0, 0); |
| 594 | |
| 595 | ASSERT_ARRAYS_EQ(keyResGold, keyRes); |
| 596 | ASSERT_ARRAYS_NEAR(valsReducedGold, valsReduced, 1e-5); |
| 597 | } |
| 598 | |
| 599 | TEST_P(ReduceByKeyP, SumDim2) { |
| 600 | if (noHalfTests(GetParam()->vType_)) { |
nothing calls this directly
no test coverage detected