Rename the class name in unittest.

PaddlePaddle · Sep 28, 2021 · daa3057 · daa3057 · paddle-bot-old · Sep 28, 2021
1 parent e94a526
commit daa3057
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/paddle/fluid/operators/fused/cudnn_norm_conv_test.cc b/paddle/fluid/operators/fused/cudnn_norm_conv_test.cc
@@ -203,9 +203,9 @@ void ComputeSumAndSquareSum(const framework::Tensor &cpu_out,
 }
 
 template <typename T>
-class TestCudnnNormConvOpForward {
+class CudnnNormConvolutionTester {
  public:
-  TestCudnnNormConvOpForward(int batch_size, int height, int width,
+  CudnnNormConvolutionTester(int batch_size, int height, int width,
                              int input_channels, int output_channels,
                              int kernel_size, int stride) {
     batch_size_ = batch_size;
@@ -219,7 +219,7 @@ class TestCudnnNormConvOpForward {
     SetUp();
   }
 
-  ~TestCudnnNormConvOpForward() {}
+  ~CudnnNormConvolutionTester() {}
 
   void CheckForward(float diff, bool is_relative_atol = false) {
     platform::CUDADeviceContext *ctx =
@@ -391,47 +391,47 @@ class TestCudnnNormConvOpForward {
 };
 
 // test for fp16, kernel = 1, output_channels = input_channels
-TEST(CudnnNormConvForward, GPUCudnnNormConvForward1Fp16) {
+TEST(CudnnNormConvFp16, K1S1) {
   int batch_size = 4;
   int height = 56;
   int width = 56;
   int input_channels = 32;
   int output_channels = 32;
   int kernel_size = 1;
   int stride = 1;
-  TestCudnnNormConvOpForward<paddle::platform::float16> test(
+  CudnnNormConvolutionTester<paddle::platform::float16> test(
       batch_size, height, width, input_channels, output_channels, kernel_size,
       stride);
   test.CheckForward(1e-3, true);
   test.CheckBackward(1e-3, true);
 }
 
 // test for fp16, kernel = 3, output_channels = input_channels
-TEST(CudnnNormConvForward, GPUCudnnNormConvForward2Fp16) {
+TEST(CudnnNormConvFp16, K3S1) {
   int batch_size = 4;
   int height = 56;
   int width = 56;
   int input_channels = 32;
   int output_channels = 32;
   int kernel_size = 3;
   int stride = 1;
-  TestCudnnNormConvOpForward<paddle::platform::float16> test(
+  CudnnNormConvolutionTester<paddle::platform::float16> test(
       batch_size, height, width, input_channels, output_channels, kernel_size,
       stride);
   test.CheckForward(1e-3, true);
   test.CheckBackward(1e-3, true);
 }
 
 // test for fp16, kernel = 1, output_channels = input_channels * 4
-TEST(CudnnNormConvForward, GPUCudnnNormConvForward3Fp16) {
+TEST(CudnnNormConvFp16, K1S1O4) {
   int batch_size = 4;
   int height = 56;
   int width = 56;
   int input_channels = 32;
   int output_channels = 128;
   int kernel_size = 1;
   int stride = 1;
-  TestCudnnNormConvOpForward<paddle::platform::float16> test(
+  CudnnNormConvolutionTester<paddle::platform::float16> test(
       batch_size, height, width, input_channels, output_channels, kernel_size,
       stride);
   test.CheckForward(1e-3, true);