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
commit
59cbf9583d
1 changed files with 0 additions and 1 deletions
  1. +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
):
if not device.physical_name.startswith("gpu"):
gopt_level = None # disable jit and compile
jit_fusion = False

if jit_fusion and not jit_supported:


Loading…
Cancel
Save