From fd40502ebf9da650795cbd88cbd50fac14f2f905 Mon Sep 17 00:00:00 2001 From: Weilong Wu Date: Tue, 7 Jun 2022 23:27:38 +0800 Subject: [PATCH] [Eager] Enable legacy for test_layout_autotune (#43273) --- python/paddle/fluid/tests/unittests/test_layout_autotune.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/python/paddle/fluid/tests/unittests/test_layout_autotune.py b/python/paddle/fluid/tests/unittests/test_layout_autotune.py index bd73d9526c0..d29f47c8ab1 100644 --- a/python/paddle/fluid/tests/unittests/test_layout_autotune.py +++ b/python/paddle/fluid/tests/unittests/test_layout_autotune.py @@ -20,6 +20,9 @@ import tempfile import warnings import json import os +from paddle.fluid.framework import _enable_legacy_dygraph + +_enable_legacy_dygraph() class SimpleNet(paddle.nn.Layer): -- GitLab