Fix additional unit tests

Author: Lunderberg

src/relax/analysis/well_formed.cc

@@ -383,7 +383,21 @@ class WellFormedChecker : public relax::ExprVisitor,
         auto inferred_struct_info = GetStructInfo(normalized.value());
         auto current_struct_info = Downcast<StructInfo>(call->struct_info_);
 
-        if (!IsBaseOf(current_struct_info, inferred_struct_info)) {
+        // An error should be raised if the annotated StructInfo is
+        // provably incorrect.  This check is done using
+        // `StructInfoBaseCheck(...) < kFailL2`, because `kFailL2`
+        // represents cases that are neither provably correct nor
+        // provably incorrect.  If this check were replaced with
+        // `!IsBaseOf(...)`, cases that are correct but not provably
+        // so would raise an exception.
+        //
+        // For example, if a dynamic size in the inferred StructInfo
+        // is equivalent to the expression used in the annotated
+        // StructInfo, but the TIR simplifications are not sufficient
+        // to prove that the two expressions are equivalent, we should
+        // not raise an error.
+        if (StructInfoBaseCheck(current_struct_info, inferred_struct_info) <
+            BaseCheckResult::kFailL2) {
           Malformed(Diagnostic::Error(call)
                     << "All information in StructInfo annotations must be correct.  "
                     << "However, while the expression " << GetRef<Call>(call) << " is annotated as "

tests/python/relax/test_ast_printer.py

@@ -366,8 +366,8 @@ def f(
     ) -> R.Object:
         m = T.int64()
         z: R.Tensor((32, m), "float32") = R.multiply(x, y)
-        w: R.Tensor = R.multiply(z, z)
-        q: R.Tensor(ndim=2) = R.add(w, w)
+        w: R.Tensor(ndim=2) = R.multiply(z, z)
+        q: R.Tensor = R.add(w, w)
         t = R.add(w, z)
         sh: R.Shape = R.shape_of(t)
         o: R.Object = R.call_packed(