Qualcomm AI Engine Direct - Fix error of avg pool2d (#11275)

chunit-quic · Joey Tsai · web-flow · commit d4906e21cda2 · 2025-06-01T22:14:52.000-07:00
- Fix the ceil_mode if condition of avg_pool2d

Co-authored-by: Joey Tsai &lt;chunit@qti.qualcomm.com&gt;
diff --git 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

Original file line number	Diff line number	Diff line change
`@@ -53,7 +53,7 @@ def define_node(`
`53`	`53`	`nodes_to_wrappers,`
`54`	`54`	`)`
`55`	`55`
`56`		`- pt_ceil_mode = node.args[4] if len(node.args) >= 4 else False`
	`56`	`+ pt_ceil_mode = node.args[4] if len(node.args) > 4 else False`
`57`	`57`
`58`	`58`	`# kernel info`
`59`	`59`	`input_shape = input_node.meta["val"].shape`