提交 6a7e7ce1 编写于 作者: M Megvii Engine Team 提交者: Xinran Xu

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

GitOrigin-RevId: 8bf00a3006599e2366204f4743855569f363ac41
上级 f19646b5
......@@ -25,7 +25,7 @@ def get_dep_vars(var, var_type=None):
outputs = []
memo = set()
if not isinstance(var, collections.Iterable):
if isinstance(var, _mgb.SymbolVar):
var = [var]
if isinstance(var_type, str):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册