提交 59cbf958 编写于 作者: M Megvii Engine Team

fix(subgraph): use CompiledOp in cpu to avoid workspace error

GitOrigin-RevId: 104dd982ef7b6a5c940bf387e440f970c9d325f2
上级 b6ce02a1
......@@ -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:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册