Qualcomm AI Engine Direct - Fix error of avg pool2d

Joey Tsai · Joey Tsai · commit 686ec42c22c6 · 2025-06-02T09:19:56.000+08:00
- Fix the ceil_mode if condition of avg_pool2d
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`