增加静态图下的 fp16 单测

PaddlePaddle · Mar 16, 2023 · 8a6ac3e · 8a6ac3e
1 parent 7dfb87c
commit 8a6ac3e
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 30 deletions.
diff --git a/python/paddle/fluid/tests/unittests/test_trapezoid.py b/python/paddle/fluid/tests/unittests/test_trapezoid.py
@@ -226,35 +226,32 @@ def test_xwithdx():
 
 
 class Testfp16Trapezoid(TestTrapezoidAPI):
-    # def test_fp16_with_gpu(self):
-    #     paddle.enable_static()
-    #     if paddle.fluid.core.is_compiled_with_cuda():
-    #         place = paddle.CUDAPlace(0)
-    #         with paddle.static.program_guard(
-    #             paddle.static.Program(), paddle.static.Program()
-    #         ):
-    #             input_y = np.random.random([4, 4]).astype("float16")
-    #             y = paddle.static.data(
-    #                 name="y", shape=[4, 4], dtype="float16"
-    #             )
-
-    #             input_x = np.random.random([4, 4]).astype("float16")
-    #             x = paddle.static.data(
-    #                 name="x", shape=[4, 4], dtype="float16"
-    #             )
-
-    #             exe = paddle.static.Executor(place)
-    #             out = self.paddle_api(y=y, x=x, dx=self.dx, axis=self.axis)
-    #             res = exe.run(
-    #                 paddle.static.default_main_program(),
-    #                 feed={
-    #                     "y": input_y,
-    #                     "x": input_x,
-    #                     "dx": self.dx,
-    #                     "axis": self.axis
-    #                 },
-    #                 fetch_list=[out],
-    #             )
+    def test_fp16_with_gpu(self):
+        paddle.enable_static()
+        if paddle.fluid.core.is_compiled_with_cuda():
+            place = paddle.CUDAPlace(0)
+            with paddle.static.program_guard(
+                paddle.static.Program(), paddle.static.Program()
+            ):
+                input_y = np.random.random([4, 4]).astype("float16")
+                y = paddle.static.data(name="y", shape=[4, 4], dtype="float16")
+
+                input_x = np.random.random([4, 4]).astype("float16")
+                x = paddle.static.data(name="x", shape=[4, 4], dtype="float16")
+
+                exe = paddle.static.Executor(place)
+                out = self.paddle_api(y=y, x=x, dx=self.dx, axis=self.axis)
+                res = exe.run(
+                    paddle.static.default_main_program(),
+                    feed={
+                        "y": input_y,
+                        "x": input_x,
+                        "dx": self.dx,
+                        "axis": self.axis,
+                    },
+                    fetch_list=[out],
+                )
+
     def set_api(self):
         self.paddle_api = paddle.trapezoid
         self.ref_api = np.trapz

diff --git a/python/paddle/tensor/math.py b/python/paddle/tensor/math.py
@@ -5286,7 +5286,6 @@ def cumulative_trapezoid(y, x=None, dx=None, axis=-1, name=None):
 
             import paddle
 
-            paddle.set_device('cpu')
             y = paddle.to_tensor([4, 5, 6], dtype='float32')
 
             print(paddle.cumulative_trapezoid(y))