[TOPI] Add shape validation to prevent negative dimensions in conv operations (#17942)

jikechao · web-flow · commit 8929d3fdd048 · 2025-05-11T11:37:39.000+08:00
* add negative output dim checking for conv2d.py

* Update conv2d.py
diff --git a/python/tvm/topi/nn/conv2d.py b/python/tvm/topi/nn/conv2d.py
@@ -743,6 +743,11 @@ def conv(
             dimensions, kernel_dimensions, dilations, pad_begin, pad_end, strides
         )
     ]
+    for out_dim in out_dimensions:
+        if isinstance(out_dim, int) and out_dim <= 0:
+            raise ValueError(
+                f"Invalid conv parameters: lead to negative output shape {out_dimensions}. "
+            )
     # compute graph
     pad_before = list(np.array([0, 0] + pad_begin)[data_permutation_from])
     pad_after = list(np.array([0, 0] + pad_end)[data_permutation_from])

Original file line number	Diff line number	Diff line change
`@@ -743,6 +743,11 @@ def conv(`
`743`	`743`	`dimensions, kernel_dimensions, dilations, pad_begin, pad_end, strides`
`744`	`744`	`)`
`745`	`745`	`]`
	`746`	`+ for out_dim in out_dimensions:`
	`747`	`+ if isinstance(out_dim, int) and out_dim <= 0:`
	`748`	`+ raise ValueError(`
	`749`	`+ f"Invalid conv parameters: lead to negative output shape {out_dimensions}. "`
	`750`	`+ )`
`746`	`751`	`# compute graph`
`747`	`752`	`pad_before = list(np.array([0, 0] + pad_begin)[data_permutation_from])`
`748`	`753`	`pad_after = list(np.array([0, 0] + pad_end)[data_permutation_from])`