diff --git a/imperative/python/megengine/core/tensor/utils.py b/imperative/python/megengine/core/tensor/utils.py index a0e2648e..97199cde 100644 --- a/imperative/python/megengine/core/tensor/utils.py +++ b/imperative/python/megengine/core/tensor/utils.py @@ -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: