未验证 提交 0e1f6cb0 编写于 作者: M MaxwellDing 提交者: GitHub

refactor: add exclude target to remove x86 nhwc kernel register (#115)

上级 aa746b71
......@@ -249,11 +249,13 @@ void OpenCLTypeLayoutTransformPass::Apply(
REGISTER_MIR_PASS(type_layout_cast_pass,
paddle::lite::mir::TypeLayoutTransformPass)
.BindTargets({TARGET(kAny)})
.ExcludeTargets({TARGET(kMLU)})
.BindKernel("layout_once")
.BindKernel("layout");
REGISTER_MIR_PASS(type_layout_cast_preprocess_pass,
paddle::lite::mir::OpenCLTypeLayoutTransformPass)
.BindTargets({TARGET(kAny)})
.ExcludeTargets({TARGET(kMLU)})
.BindKernel("layout_once")
.BindKernel("layout");
......@@ -184,9 +184,6 @@ KernelRegistry::KernelRegistry()
INIT_FOR(kX86, kFloat, kNCHW);
INIT_FOR(kX86, kFP16, kNCHW);
INIT_FOR(kX86, kInt8, kNCHW);
INIT_FOR(kX86, kFloat, kNHWC);
INIT_FOR(kX86, kFP16, kNHWC);
INIT_FOR(kX86, kInt8, kNHWC);
INIT_FOR(kX86, kAny, kNCHW);
INIT_FOR(kX86, kAny, kAny);
INIT_FOR(kX86, kInt64, kNCHW);
......
......@@ -126,15 +126,6 @@ class KernelRegistry final {
KernelRegistryForTarget<TARGET(kX86),
PRECISION(kInt8),
DATALAYOUT(kNCHW)> *, //
KernelRegistryForTarget<TARGET(kX86),
PRECISION(kFloat),
DATALAYOUT(kNHWC)> *, //
KernelRegistryForTarget<TARGET(kX86),
PRECISION(kFP16),
DATALAYOUT(kNHWC)> *, //
KernelRegistryForTarget<TARGET(kX86),
PRECISION(kInt8),
DATALAYOUT(kNHWC)> *, //
KernelRegistryForTarget<TARGET(kHost),
PRECISION(kFloat),
DATALAYOUT(kNCHW)> *, //
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册