未验证 提交 7e27b7bf 编写于 作者: Z zhupengyang 提交者: GitHub

[XPU] not run x86 kernels for xpu (#2373)

test=develop
上级 b44623c5
......@@ -229,6 +229,8 @@ void SubgraphProgramPass::InferOnce(const std::unique_ptr<SSAGraph>& graph) {
}
op->CheckShape();
op->InferShape();
#ifndef LITH_WITH_XPU
// TOOD(xxx): remove Launch() at last
auto& kkks = stmt.kernels();
if (!kkks.empty()) {
......@@ -237,6 +239,7 @@ void SubgraphProgramPass::InferOnce(const std::unique_ptr<SSAGraph>& graph) {
kk->Launch();
}
}
#endif
}
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册