Skip to content

Commit 6f9b930

Browse files
authored
[HuaweiAscendNPU] Fix adaptive_max_pool2d (#10140)
1 parent 62ea259 commit 6f9b930

File tree

2 files changed

+1
-7
lines changed

2 files changed

+1
-7
lines changed

lite/backends/nnadapter/nnadapter/src/driver/huawei_ascend_npu/converter/adaptive_pool2d.cc

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,6 @@ int ConvertAdaptivePool2D(Converter* converter, core::Operation* operation) {
3030
input_operator = converter->ConvertOperand(input_operand);
3131
}
3232
if (operation->type == NNADAPTER_ADAPTIVE_MAX_POOL_2D) {
33-
#if NNADAPTER_HUAWEI_ASCEND_NPU_CANN_VERSION_LESS_THAN(5, 1, 1)
3433
auto pool2d_op =
3534
converter->AddOperator<ge::op::AdaptiveMaxPool2d>(output_operand);
3635
pool2d_op->set_attr_output_size(
@@ -60,10 +59,6 @@ int ConvertAdaptivePool2D(Converter* converter, core::Operation* operation) {
6059
SET_INPUT(dummy_add_op, x1, adaptive_pool2d_operator);
6160
SET_INPUT(dummy_add_op, x2, dummy_sub_operator);
6261
MAP_OUTPUT(dummy_add_op, y, output_operand);
63-
#else
64-
NNADAPTER_LOG(FATAL) << "AdaptiveMaxPool2d has bugs when CANN >= 5.1.rc1, "
65-
"it will be fixed later";
66-
#endif
6762
} else {
6863
auto pool2d_op =
6964
converter->AddOperator<ge::op::AdaptiveAvgPool2d>(output_operand);

lite/tests/kernels/adaptive_pool_compute_test.cc

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ void TestAdaptiveAveragePool2D(Place place, float abs_error = 2e-5) {
154154
}
155155

156156
void TestAdaptiveMaxPool2D(Place place, float abs_error = 2e-5) {
157-
for (auto dims : std::vector<std::vector<int64_t>>{{2, 3, 4, 5}}) {
157+
for (auto dims : std::vector<std::vector<int64_t>>{{2, 3, 6, 6}}) {
158158
std::unique_ptr<arena::TestCase> tester(
159159
new AdaptivePoolTester(place, "def", DDim(dims), "max", false));
160160
arena::Arena arena(std::move(tester), place, abs_error);
@@ -192,7 +192,6 @@ TEST(AdaptiveMaxPool2D, precision) {
192192
place = TARGET(kNNAdapter);
193193
#if defined(NNADAPTER_WITH_HUAWEI_ASCEND_NPU)
194194
abs_error = 1e-2;
195-
return;
196195
#else
197196
return;
198197
#endif

0 commit comments

Comments
 (0)