This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
Repositories
Datasets
Forum
实训
竞赛
大数据
应用
Register
Sign In
OpenI
/
MegEngine
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
31
Wiki
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
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
f19646b51f
commit
6a7e7ce1b1
1 changed files
with
1 additions
and
1 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+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 = []
memo = set()
if
not isinstance(var, collections.Iterable
):
if
isinstance(var, _mgb.SymbolVar
):
var = [var]
if isinstance(var_type, str):
Write
Preview
Loading…
Cancel
Save