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