From 686ec42c22c6528d64dc9f274c9ed083d139f17f Mon Sep 17 00:00:00 2001 From: Joey Tsai Date: Mon, 2 Jun 2025 09:10:26 +0800 Subject: [PATCH] Qualcomm AI Engine Direct - Fix error of avg pool2d - Fix the ceil_mode if condition of avg_pool2d --- backends/qualcomm/builders/op_avg_pool2d.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backends/qualcomm/builders/op_avg_pool2d.py b/backends/qualcomm/builders/op_avg_pool2d.py index bd7094cc382..6892e7326f6 100644 --- a/backends/qualcomm/builders/op_avg_pool2d.py +++ b/backends/qualcomm/builders/op_avg_pool2d.py @@ -53,7 +53,7 @@ def define_node( nodes_to_wrappers, ) - pt_ceil_mode = node.args[4] if len(node.args) >= 4 else False + pt_ceil_mode = node.args[4] if len(node.args) > 4 else False # kernel info input_shape = input_node.meta["val"].shape