[Test][XPU] fix typos in test_elementwise_add_op (#63898)

dynamicheart · web-flow · commit 0e58487131c2 · 2024-04-30T14:57:26.000+08:00
diff --git a/test/legacy_test/test_elementwise_add_op.py b/test/legacy_test/test_elementwise_add_op.py
@@ -880,8 +880,8 @@ def test_warnings(self):
             os.environ['FLAGS_print_extra_attrs'] = "0"
 
 
-class TestTensorFloa32Bfloat16OrFloat16Add(unittest.TestCase):
-    def _floa32_bfloat16_or_float16_add(self, y_dtype):
+class TestTensorFloat32Bfloat16OrFloat16Add(unittest.TestCase):
+    def _float32_bfloat16_or_float16_add(self, y_dtype):
         paddle.disable_static()
         test_num = 5
         val_range = 10000
@@ -906,22 +906,22 @@ def _floa32_bfloat16_or_float16_add(self, y_dtype):
     or paddle.device.cuda.get_device_capability()[0] < 8,
     "only support compiled with CUDA and cudnn version need larger than 8.1.0 and device's compute capability is at least 8.0",
 )
-class TestTensorFloa32Bfloat16Add(TestTensorFloa32Bfloat16OrFloat16Add):
-    def test_floa32_bfloat16_add(self):
+class TestTensorFloat32Bfloat16Add(TestTensorFloat32Bfloat16OrFloat16Add):
+    def test_float32_bfloat16_add(self):
         place = core.CUDAPlace(0)
         with base.dygraph.base.guard(place=place):
-            self._floa32_bfloat16_or_float16_add(y_dtype=paddle.bfloat16)
+            self._float32_bfloat16_or_float16_add(y_dtype=paddle.bfloat16)
 
 
 @unittest.skipIf(
     not core.is_compiled_with_cuda() or core.cudnn_version() < 8100,
     "only support compiled with CUDA and cudnn version need larger than 8.1.0",
 )
-class TestTensorFloa32Float16Add(TestTensorFloa32Bfloat16OrFloat16Add):
-    def test_floa32_float16_add(self):
+class TestTensorFloat32Float16Add(TestTensorFloat32Bfloat16OrFloat16Add):
+    def test_float32_float16_add(self):
         place = core.CUDAPlace(0)
         with base.dygraph.base.guard(place=place):
-            self._floa32_bfloat16_or_float16_add(y_dtype=paddle.float16)
+            self._float32_bfloat16_or_float16_add(y_dtype=paddle.float16)
 
 
 class TestElementwiseAddOpAutoParallel(OpTest):
diff --git a/test/xpu/test_elementwise_add_op_xpu.py b/test/xpu/test_elementwise_add_op_xpu.py
@@ -310,8 +310,8 @@ def test_dygraph(self):
     core.get_xpu_device_version(0) != core.XPUVersion.XPU3,
     "only supported on XPU3",
 )
-class TestTensorFloa32Bfloat16OrFloat16Add(unittest.TestCase):
-    def _floa32_bfloat16_or_float16_add(self, y_dtype):
+class TestTensorFloat32Bfloat16OrFloat16Add(unittest.TestCase):
+    def _float32_bfloat16_or_float16_add(self, y_dtype):
         paddle.disable_static()
         test_num = 5
         val_range = 10000
@@ -329,11 +329,11 @@ def _floa32_bfloat16_or_float16_add(self, y_dtype):
             np.testing.assert_equal(x.numpy(), x_copy.numpy())
             del x, x_copy
 
-    def test_floa32_bfloat16_add(self):
-        self._floa32_bfloat16_or_float16_add(y_dtype=paddle.bfloat16)
+    def test_float32_bfloat16_add(self):
+        self._float32_bfloat16_or_float16_add(y_dtype=paddle.bfloat16)
 
-    def test_floa32_float16_add(self):
-        self._floa32_bfloat16_or_float16_add(y_dtype=paddle.float16)
+    def test_float32_float16_add(self):
+        self._float32_bfloat16_or_float16_add(y_dtype=paddle.float16)
 
 
 if __name__ == '__main__':
