speed up

kendryte · Dec 29, 2024 · 4d27d7d · 4d27d7d
1 parent 625bfa1
commit 4d27d7d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/Nncase.Core/IR/Buffers/DDrOf.cs b/src/Nncase.Core/IR/Buffers/DDrOf.cs
@@ -16,7 +16,7 @@ public sealed partial class DDrOf : Op
     /// <summary>
     /// Get the input parameter.
     /// </summary>
-    public static readonly ParameterInfo Input = new(typeof(DDrOf), 0, "input", IsTensor());
+    public static readonly ParameterInfo Input = new(typeof(DDrOf), 0, "input", IsTensor() | IsPointer());
 
     /// <inheritdoc/>
     public override bool CanFoldConstCall => false;

diff --git a/src/Nncase.Passes/Rules/ShapeBucket/RecordFusionShape.cs b/src/Nncase.Passes/Rules/ShapeBucket/RecordFusionShape.cs
@@ -233,7 +233,7 @@ protected override Task<BaseFunction> RunCoreAsync(BaseFunction main, RunPassCon
             {
                 var varValues = seg.ToDictionary(pair => pair.Key, pair => (IValue)Value.FromTensor(pair.Value));
                 var exprValues = seg.ToDictionary(pair => (Expr)pair.Key, pair => (IValue)Value.FromTensor(pair.Value));
-#if true
+#if false
                 var input = MakeDummyInput(varMap, varValues);
                 var memo = EvaluatorUtil.GetMemo(body, ConcatDictionary(input, varValues));
                 var f = new FusionShapeUpdater(ConcatDictionary(memo, exprValues));