From df53f7f0a3af01e1fac4bdcfb6bd4f5dddc39548 Mon Sep 17 00:00:00 2001 From: Wang Huan Date: Sun, 27 Sep 2020 07:35:57 +0000 Subject: [PATCH] fix bug about core_avx core_noavx, test=develop --- python/paddle/fluid/__init__.py | 3 ++- python/paddle/fluid/core.py | 2 ++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/python/paddle/fluid/__init__.py b/python/paddle/fluid/__init__.py index 9e60044a520..881fb86266f 100644 --- a/python/paddle/fluid/__init__.py +++ b/python/paddle/fluid/__init__.py @@ -90,7 +90,8 @@ from .io import save, load, load_program_state, set_program_state from .dygraph.checkpoint import save_dygraph, load_dygraph from .dygraph.varbase_patch_methods import monkey_patch_varbase from . import generator -from .core_avx import _cuda_synchronize +from .core import _cuda_synchronize + Tensor = LoDTensor enable_imperative = enable_dygraph disable_imperative = disable_dygraph diff --git a/python/paddle/fluid/core.py b/python/paddle/fluid/core.py index 2e3bb6b0021..9a14c4cdf14 100644 --- a/python/paddle/fluid/core.py +++ b/python/paddle/fluid/core.py @@ -263,6 +263,7 @@ if avx_supported(): from .core_avx import _save_dygraph_dict from .core_avx import _load_dygraph_dict from .core_avx import _create_loaded_parameter + from .core_avx import _cuda_synchronize if sys.platform != 'win32': from .core_avx import _set_process_pids from .core_avx import _erase_process_pids @@ -307,6 +308,7 @@ if load_noavx: from .core_noavx import _save_dygraph_dict from .core_noavx import _load_dygraph_dict from .core_noavx import _create_loaded_parameter + from .core_noavx import _cuda_synchronize if sys.platform != 'win32': from .core_noavx import _set_process_pids from .core_noavx import _erase_process_pids -- GitLab