Browse Source

fix(mge/cgtools): make get_dep_vars accept symbolvar input correctly

GitOrigin-RevId: 8bf00a3006
tags/v0.5.0
Megvii Engine Team Xinran Xu 5 years ago
parent
commit
6a7e7ce1b1
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      python_module/megengine/_internal/comp_graph_tools.py

+ 1
- 1
python_module/megengine/_internal/comp_graph_tools.py View File

@@ -25,7 +25,7 @@ def get_dep_vars(var, var_type=None):
outputs = [] outputs = []
memo = set() memo = set()


if not isinstance(var, collections.Iterable):
if isinstance(var, _mgb.SymbolVar):
var = [var] var = [var]


if isinstance(var_type, str): if isinstance(var_type, str):


Loading…
Cancel
Save