| 36 | class RemapperTest : public GrapplerTest { |
| 37 | protected: |
| 38 | void SetUp() override { |
| 39 | // This is a requirement for fusing FusedBatchNorm + SideInput + Activation. |
| 40 | setenv("TF_USE_CUDNN_BATCHNORM_SPATIAL_PERSISTENT", "1", 1 /* replace */); |
| 41 | // This is a requirement for fusing FusedMatmul + BiasAdd (+ Activation). |
| 42 | setenv("TF_USE_CUBLASLT", "1", 1 /* replace */); |
| 43 | } |
| 44 | }; |
| 45 | |
| 46 | TEST_F(RemapperTest, FusedBatchNorm) { |