import queue
import threading
import functools
import paddle
from paddle.base.framework import global_var
print('-'*3, 'main thread in ...', global_var._dygraph_tracer_, global_var._functional_dygraph_context_manager)
paddle.disable_static()
# paddle.enable_static()
print('paddle.in_dynamic_mode', paddle.in_dynamic_mode())
print('global_var._dygraph_tracer_', global_var._dygraph_tracer_, global_var._functional_dygraph_context_manager)
print('-'*3, 'main thread out ...', global_var._dygraph_tracer_, global_var._functional_dygraph_context_manager)
def test():
    import paddle
    from paddle.base.framework import global_var
    print('-'*3, 'sub thread in ...', global_var._dygraph_tracer_, global_var._functional_dygraph_context_manager)
    paddle.enable_static()
    print('paddle.in_dynamic_mode', paddle.in_dynamic_mode()) # 此处应该为 `False` ，但实际为 `True`
    print('global_var._dygraph_tracer_', global_var._dygraph_tracer_, global_var._functional_dygraph_context_manager)
    print('-'*3, 'sub thread out ...', global_var._dygraph_tracer_, global_var._functional_dygraph_context_manager)
    data = paddle.static.data(name='X', shape=[None, 2, 28, 28], dtype='float32')
    return data
result_queue = queue.Queue()
exec_processer = functools.partial(threading.Thread, daemon=True)
def _execute_with_queue(queue):
    queue.put(test())
processer = exec_processer(
    target=_execute_with_queue,
    args=(result_queue,)
)
processer.start()
result = result_queue.get(timeout=100)
processer.join()
print(result)

# use thread local to create thread save global variables.
class GlobalThreadLocal(threading.local):
    def __init__(self):
        """
        init the thread local data.
        TODO(xiongkun): how to access another thread local data ?
        """
        global _dygraph_tracer_
        self._in_to_static_mode_ = False
        self._functional_dygraph_context_manager = None
        self._dygraph_tracer_ = _dygraph_tracer_
        self._use_pir_api_ = get_flags("FLAGS_enable_pir_api")[
            "FLAGS_enable_pir_api"
        ]
    def __str__(self):
        strings = []
        strings.append("_in_to_static_mode_:" + str(self._in_to_static_mode_))
        strings.append(
            "_functional_dygraph_context_manager:"
            + str(self._functional_dygraph_context_manager)
        )
        strings.append("_dygraph_tracer_:" + str(self._dygraph_tracer_))
        return "\n".join(strings)
    def __setattr__(self, name, val):
        if name == "_dygraph_tracer_":
            global _dygraph_tracer_
            _dygraph_tracer_ = val
            core._switch_tracer(val)
        self.__dict__[name] = val
_dygraph_tracer_ = None
global_var = GlobalThreadLocal()