[pir] delete set input_spec as none in dynamic shape in input_spec (#68444)

xiaoguoguo626807 · web-flow · commit 548ee67532cd · 2024-09-27T14:05:52.000+08:00
* modify paddletest

* modify paddletest

* remove dynamic shape

* modify ci
diff --git a/python/paddle/jit/dy2static/program_translator.py b/python/paddle/jit/dy2static/program_translator.py
@@ -68,7 +68,6 @@
     input_specs_compatible,
     is_paddle_func,
     make_hashable,
-    prim_is_enabled,
     prim_or_cinn_is_enabled,
     type_name,
 )
@@ -402,21 +401,6 @@ def __init__(self, function, input_spec=None, **kwargs):
         else:
             self._dygraph_function = function
             self._class_instance = None
-        # TODO(chenzhuo): Remove this after lowering prim into C++
-        if (
-            input_spec is not None
-            and prim_is_enabled()
-            and not core._enable_prim_dynamic_shape()
-        ):
-            from paddle.static import InputSpec
-
-            for spec in flatten(input_spec):
-                if isinstance(spec, InputSpec) and -1 in spec.shape:
-                    input_spec = None
-                    warnings.warn(
-                        'Now prim and cinn do not support -1 shape, but input_spec has -1 shape so we set it to None.'
-                    )
-                    break
 
         self._input_spec = input_spec
         self._function_spec = FunctionSpec(function, input_spec)
diff --git a/test/legacy_test/test_input_spec.py b/test/legacy_test/test_input_spec.py
@@ -370,7 +370,7 @@ def test_run(self):
         )
         x = paddle.randn([2, 10])
         out = net(x)
-        np.testing.assert_equal(net.forward._input_spec, None)
+        np.testing.assert_equal(out.shape, [2, 5])
 
 
 if __name__ == '__main__':

Original file line number	Diff line number	Diff line change
`@@ -370,7 +370,7 @@ def test_run(self):`
`370`	`370`	`)`
`371`	`371`	`x = paddle.randn([2, 10])`
`372`	`372`	`out = net(x)`
`373`		`- np.testing.assert_equal(net.forward._input_spec, None)`
	`373`	`+ np.testing.assert_equal(out.shape, [2, 5])`
`374`	`374`
`375`	`375`
`376`	`376`	`if __name__ == '__main__':`