[Transform] Remove R.Object parameters after LazyTranformParams

Prior to this commit, the output of
`relax.transform.LazyTransformParams` would include all parameters
that are not `R.Tensor`, in case they defined symbolic variables.
However, this added too many unnecessary parameters, such as
`R.Object` which cannot define symbolic variables.  This commit
updates `relax.transform.LazyTransformParams` to only retain `R.Prim`
and `R.Shape` parameters, which can define symbolic variables.

Author: Lunderberg

python/tvm/relax/transform/lazy_transform_params.py

@@ -216,7 +216,9 @@ def unpack_sinfo(sinfo):
             # direct iterate over the struct info annotation
             for param in func.params[num_input:]:
                 for sinfo in unpack_sinfo(param.struct_info):
-                    if not isinstance(sinfo, relax.TensorStructInfo):
+                    if isinstance(sinfo, relax.PrimStructInfo) or isinstance(
+                        sinfo, relax.ShapeStructInfo
+                    ):
                         params.append(relax.Var("symbolic_var_holder", sinfo))
 
         return relax.Function(

tests/python/relax/test_transform_lazy_transform_params.py

@@ -804,5 +804,25 @@ def transform_params(relax_rank: R.Prim(value="rank")):
     tvm.ir.assert_structural_equal(After, Expected)
 
 
+def test_params_without_tuple_with_symbolic_var():
+    @I.ir_module
+    class Before:
+        @R.function
+        def transform_params(A: R.Object):
+            return (A,)
+
+    @I.ir_module
+    class Expected:
+        @R.function(pure=False)
+        def transform_params():
+            A = R.call_packed("get_item", R.prim_value(0), sinfo_args=[R.Object])
+            A = R.match_cast(A, R.Object)
+
+            return (A,)
+
+    After = LazyTransformParams(fset_item=None)(Before)
+    tvm.ir.assert_structural_equal(After, Expected)
+
+
 if __name__ == "__main__":
     tvm.testing.main()