From b2f9efb8971f8cb1acb4224102ff0cbd6196e720 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Sun, 27 Sep 2020 16:13:57 +0800 Subject: [PATCH] perf(mge): always async dispatch in jit.trace GitOrigin-RevId: 6fa80b0dd9e545cc1789de02138456b1cfa8d74c --- imperative/python/megengine/jit/tracing.py | 1 + 1 file changed, 1 insertion(+) diff --git a/imperative/python/megengine/jit/tracing.py b/imperative/python/megengine/jit/tracing.py index 842fef52..1eab184f 100644 --- a/imperative/python/megengine/jit/tracing.py +++ b/imperative/python/megengine/jit/tracing.py @@ -368,6 +368,7 @@ class trace: def _compile(self): graph = self._graph = G.Graph() graph.options.no_force_inplace = True + graph.options.async_exec_level = 0b100 self._apply_graph_options(graph) # graph.options.graph_opt_level = 0 need_reset_nodes = self._need_reset_nodes = []