From 59cbf9583d0259e199b75d4a026f6dba735c9d16 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Fri, 3 Dec 2021 12:54:13 +0800
Subject: [PATCH] fix(subgraph): use CompiledOp in cpu to avoid workspace error

GitOrigin-RevId: 104dd982ef7b6a5c940bf387e440f970c9d325f2
---
 imperative/python/megengine/core/tensor/utils.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/imperative/python/megengine/core/tensor/utils.py b/imperative/python/megengine/core/tensor/utils.py
index a0e2648eb..97199cde9 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:
-- 
GitLab