Skip to content

Commit 62ea259

Browse files
authored
[HuaweiAscendNPU] Convert AdaptivePool2d into Pool2d when CANN > 6.0.0 (#10143)
1 parent dc9c1ad commit 62ea259

File tree

1 file changed

+4
-0
lines changed
  • lite/backends/nnadapter/nnadapter/src/driver/huawei_ascend_npu

1 file changed

+4
-0
lines changed

lite/backends/nnadapter/nnadapter/src/driver/huawei_ascend_npu/engine.cc

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
#include "driver/huawei_ascend_npu/optimizer/fix_no_inputs_ops.h"
1919
#include "driver/huawei_ascend_npu/optimizer/fix_quantized_ops.h"
2020
#include "driver/huawei_ascend_npu/optimizer/fix_reduce_ops_scalar_output.h"
21+
#include "optimizer/convert_adaptive_pool2d_into_pool2d.h"
2122
#include "optimizer/fuse_conv2d_activation_into_conv2d.h"
2223
#include "optimizer/fuse_conv2d_add_into_conv2d.h"
2324
#include "optimizer/fuse_conv2d_batch_norm_into_conv2d.h"
@@ -238,6 +239,9 @@ int Program::Build(core::Model* model, core::Cache* cache) {
238239
FuseMatMulDequantAddIntoFullyConnectedDequant(model);
239240
FuseMatMulAddIntoFullyConnected(model, true);
240241
FuseReshapeTransposeReshapeIntoChannelShuffle(model);
242+
#if NNADAPTER_HUAWEI_ASCEND_NPU_CANN_VERSION_GREATER_THAN(6, 0, 0)
243+
ConvertAdaptivePool2dIntoPool2d(model);
244+
#endif
241245
FixQuantizedOps(model);
242246
NNADAPTER_VLOG(5) << "Optimized model:" << std::endl << Visualize(model);
243247
// Convert a NNAdapter model to a GE graph

0 commit comments

Comments
 (0)