From 68f47d84a419207df21cf6dceef59307ed921c6e Mon Sep 17 00:00:00 2001 From: Weilong Wu Date: Mon, 25 Apr 2022 10:35:34 +0800 Subject: [PATCH] [Eager] Remove redundancy code, fix fp16 case (#42169) --- python/paddle/fluid/initializer.py | 1 - .../fluid/tests/unittests/test_dygraph_mnist_fp16.py | 8 +++++++- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/initializer.py b/python/paddle/fluid/initializer.py index ab38bbf56ee3c..5177d752a15fe 100644 --- a/python/paddle/fluid/initializer.py +++ b/python/paddle/fluid/initializer.py @@ -353,7 +353,6 @@ def __call__(self, var, block=None): out_var = _C_ops.final_state_gaussian_random( var.shape, self._mean, self._std_dev, self._seed, out_dtype, place) - out_var._share_underline_tensor_to(var) if var.dtype in [VarDesc.VarType.FP16, VarDesc.VarType.BF16]: var_tmp = _C_ops.final_state_cast(out_var, var.dtype) diff --git a/python/paddle/fluid/tests/unittests/test_dygraph_mnist_fp16.py b/python/paddle/fluid/tests/unittests/test_dygraph_mnist_fp16.py index 7503a9172fc21..6c2516d6c11ef 100644 --- a/python/paddle/fluid/tests/unittests/test_dygraph_mnist_fp16.py +++ b/python/paddle/fluid/tests/unittests/test_dygraph_mnist_fp16.py @@ -19,6 +19,7 @@ import paddle.fluid as fluid from paddle.fluid.dygraph.nn import Conv2D, Pool2D, Linear +from paddle.fluid.framework import _test_eager_guard class SimpleImgConvPool(fluid.dygraph.Layer): @@ -117,7 +118,7 @@ def forward(self, inputs, label): class TestMnist(unittest.TestCase): - def test_mnist_fp16(self): + def func_mnist_fp16(self): if not fluid.is_compiled_with_cuda(): return x = np.random.randn(1, 3, 224, 224).astype("float16") @@ -129,6 +130,11 @@ def test_mnist_fp16(self): loss = model(x, y) print(loss.numpy()) + def test_mnist_fp16(self): + with _test_eager_guard(): + self.func_mnist_fp16() + self.func_mnist_fp16() + if __name__ == "__main__": unittest.main()