| 222 | } |
| 223 | |
| 224 | XLA_TEST_P(GroupedConvolution2DTest, DoIt) { |
| 225 | const GroupedConvolution2DSpec& spec = ::testing::get<0>(GetParam()); |
| 226 | bool use_bfloat16 = ::testing::get<1>(GetParam()); |
| 227 | |
| 228 | #ifdef XLA_BACKEND_DOES_NOT_SUPPORT_BFLOAT16 |
| 229 | if (use_bfloat16) { |
| 230 | return; |
| 231 | } |
| 232 | #endif |
| 233 | |
| 234 | const string hlo_text = BuildHloTextGroupedConvolution2D(spec, use_bfloat16); |
| 235 | |
| 236 | EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{0.01, 0.01}, |
| 237 | [](HloModule* module) -> Status { |
| 238 | BFloat16MixedPrecisionRemoval remover; |
| 239 | TF_RETURN_IF_ERROR(remover.Run(module).status()); |
| 240 | Despecializer despecializer; |
| 241 | return despecializer.Run(module).status(); |
| 242 | })); |
| 243 | } |
| 244 | |
| 245 | INSTANTIATE_TEST_CASE_P( |
| 246 | GroupedConvolution2DTestWithRandomIndices, GroupedConvolution2DTest, |
nothing calls this directly
no test coverage detected