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(subgraph): use CompiledOp in cpu to avoid workspace error
GitOrigin-RevId:
104dd982ef
tags/v1.9.0
Megvii Engine Team
3 years ago
parent
b6ce02a152
commit
59cbf9583d
1 changed files
with
0 additions
and
1 deletions
Unified View
Diff Options
Show Stats
Download Patch File
Download Diff File
+0
-1
imperative/python/megengine/core/tensor/utils.py
+ 0
- 1
imperative/python/megengine/core/tensor/utils.py
View File
@@ -232,7 +232,6 @@ def subgraph(
name, dtype, device, nr_inputs, gopt_level=None, jit_fusion=False, custom_grad=False
name, dtype, device, nr_inputs, gopt_level=None, jit_fusion=False, custom_grad=False
):
):
if not device.physical_name.startswith("gpu"):
if not device.physical_name.startswith("gpu"):
gopt_level = None # disable jit and compile
jit_fusion = False
jit_fusion = False
if jit_fusion and not jit_supported:
if jit_fusion and not jit_supported:
Write
Preview
Loading…
Cancel
Save