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