From 7e27b7bf587e81baf4a2fa9f5c95077bfa2d3540 Mon Sep 17 00:00:00 2001 From: zhupengyang Date: Tue, 5 Nov 2019 22:39:54 +0800 Subject: [PATCH] [XPU] not run x86 kernels for xpu (#2373) test=develop --- lite/core/mir/subgraph/subgraph_program_pass.cc | 3 +++ 1 file changed, 3 insertions(+) diff --git a/lite/core/mir/subgraph/subgraph_program_pass.cc b/lite/core/mir/subgraph/subgraph_program_pass.cc index 0cb2261a3f..719a01dfd8 100644 --- a/lite/core/mir/subgraph/subgraph_program_pass.cc +++ b/lite/core/mir/subgraph/subgraph_program_pass.cc @@ -229,6 +229,8 @@ void SubgraphProgramPass::InferOnce(const std::unique_ptr& 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& graph) { kk->Launch(); } } +#endif } } -- GitLab