Print dim with script

Author: sunnycase

src/Nncase.Core/IR/Expr.cs

@@ -47,6 +47,8 @@ internal Expr(IEnumerable<Expr> operands)
         {
             operand.AddUser(this);
         }
+
+        RefreshDepth();
     }
 
     internal Expr(Expr[] operands)
@@ -57,8 +59,12 @@ internal Expr(Expr[] operands)
         {
             operand.AddUser(this);
         }
+
+        RefreshDepth();
     }
 
+    public int Depth { get; private set; }
+
     public IRMetadata Metadata { get; set; } = new();
 
     /// <summary>
@@ -318,6 +324,7 @@ private void OnOperandsReplaced()
         InvalidateTypeInference();
         InvalidateHashCodeCache();
         InvalidateRange();
+        RefreshDepth();
     }
 
     private void InvalidateTypeInference()
@@ -370,4 +377,9 @@ private void InvalidateUsersRange()
             user.InvalidateRange();
         }
     }
+
+    private void RefreshDepth()
+    {
+        Depth = _operands.Length == 0 ? 0 : _operands.Max(x => x.Depth) + 1;
+    }
 }

src/Nncase.Diagnostics/Diagnostics/ILPrintVisitor.cs

@@ -721,7 +721,7 @@ private string VisitDimension(Dimension dimension) =>
         {
             DimensionKind.Any => "any",
             DimensionKind.Fixed => dimension.FixedValue.ToString(),
-            DimensionKind.Unknown => dimension.Value is Var var ? $"%{var.Name}" : "?",
+            DimensionKind.Unknown => dimension.Value is Var var ? $"%{var.Name}" : (dimension.Value.Depth > 4 ? "..." : CompilerServices.Print(dimension.Value, true)),
             _ => throw new NotSupportedException(dimension.Kind.ToString()),
         };
 

src/Nncase.Diagnostics/Diagnostics/ScriptPrintVisitor.cs

@@ -324,6 +324,44 @@ protected override IPrintSymbol VisitCall(Call expr)
         return doc;
     }
 
+    /// <inheritdoc/>
+    protected override IPrintSymbol VisitIf(If expr)
+    {
+        if (_exprMemo.TryGetValue(expr, out var doc))
+        {
+            return doc;
+        }
+
+        doc = new(new($"if ..."));
+        _exprMemo.Add(expr, doc!);
+        return doc!;
+
+        // var target = Visit(expr.Target);
+        // var args = expr.Arguments.AsValueEnumerable().Select(Visit).ToArray();
+        // _context.CurrentCall = expr;
+        // _scope.Push();
+        // switch (expr.Target)
+        // {
+        //     case Op op:
+        //         _scope.Append(CompilerServices.PrintOp(op, _context, false));
+        //         break;
+        //     case Function:
+        //         _scope.Append($"{target.Name}({string.Join(", ", from a in args select a.ToString())})");
+        //         break;
+        //     case TIR.PrimFunction:
+        //         _scope.AppendLine(string.Empty);
+        //         _scope.IndWrite($"{target.Name}({string.Join(", ", from a in args select a.ToString())})");
+        //         break;
+        //     default:
+        //         _scope.Append($"{target}({string.Join(", ", from a in args select a.ToString())})");
+        //         break;
+        // }
+
+        // doc = new(_scope.Pop());
+        // _exprMemo.Add(expr, doc);
+        // return doc;
+    }
+
     /// <inheritdoc/>
     protected override IPrintSymbol VisitTensorConst(TensorConst @const)
     {