Browse Source

perf(mge): always async dispatch in jit.trace

GitOrigin-RevId: 6fa80b0dd9
release-1.1
Megvii Engine Team 4 years ago
parent
commit
b2f9efb897
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      imperative/python/megengine/jit/tracing.py

+ 1
- 0
imperative/python/megengine/jit/tracing.py View File

@@ -368,6 +368,7 @@ class trace:
def _compile(self): def _compile(self):
graph = self._graph = G.Graph() graph = self._graph = G.Graph()
graph.options.no_force_inplace = True graph.options.no_force_inplace = True
graph.options.async_exec_level = 0b100
self._apply_graph_options(graph) self._apply_graph_options(graph)
# graph.options.graph_opt_level = 0 # graph.options.graph_opt_level = 0
need_reset_nodes = self._need_reset_nodes = [] need_reset_nodes = self._need_reset_nodes = []


Loading…
Cancel
Save