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

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

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