Browse Source

disable fp32 sliding window kernel

tags/v0.7.0-beta
lixian 5 years ago
parent
commit
8ab805d6ce
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      mindspore/lite/src/runtime/kernel/arm/fp32/convolution.cc

+ 2
- 1
mindspore/lite/src/runtime/kernel/arm/fp32/convolution.cc View File

@@ -258,7 +258,8 @@ kernel::LiteKernel *CpuConvFp32KernelCreator(const std::vector<lite::tensor::Ten
kernel =
new (std::nothrow) kernel::ConvolutionWinogradCPUKernel(op_parameter, inputs, outputs, ctx, primitive, out_unit);
} else if (use_sw) {
kernel = new (std::nothrow) kernel::ConvolutionSWCPUKernel(op_parameter, inputs, outputs, ctx, primitive);
// kernel = new (std::nothrow) kernel::ConvolutionSWCPUKernel(op_parameter, inputs, outputs, ctx, primitive);
kernel = new (std::nothrow) kernel::ConvolutionCPUKernel(op_parameter, inputs, outputs, ctx, primitive);
} else {
kernel = new (std::nothrow) kernel::ConvolutionCPUKernel(op_parameter, inputs, outputs, ctx, primitive);
}


Loading…
Cancel
Save