GNNE-1714:Fix/search time egraph (#1155)

* change name method for getitem for lstm * change search time to 120s --------- Co-authored-by: guodongliang <guodongliang@canaan-creative.com>
kendryte · Jan 5, 2024 · ebac64a · ebac64a
1 parent 52fe13b
commit ebac64a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/Nncase.EGraph/Passes/EGraphExtractors/SatExtractor.cs b/src/Nncase.EGraph/Passes/EGraphExtractors/SatExtractor.cs
@@ -77,7 +77,7 @@ public Expr Extract(EClass root, IEGraph eGraph, out IReadOnlyDictionary<ENode,
         }
 
         var solver = new CpSolver();
-        int max_time = 600;
+        int max_time = 120;
         if (System.Environment.GetEnvironmentVariable("SOLVE_MAX_TIME") is string s_solve_max_time)
         {
             try

diff --git a/src/Nncase.Passes/Rules/Neutral/AddRangeOfAndMarker.cs b/src/Nncase.Passes/Rules/Neutral/AddRangeOfAndMarker.cs
@@ -195,7 +195,7 @@ private IR.Tuple WrapLSTMOutput(Call call, int outputSize, bool configExist, boo
         {
             var outputNames = new List<string>();
             var getItem = IR.F.Tensors.GetItem(call, i);
-            outputNames.Add("LSTMOutput_" + call.Metadata.OutputNames?[i]);
+            outputNames.Add(call.Metadata.OutputNames?[i] ?? "LSTMOutput_" + i.ToString());
             outputs[i].Metadata.OutputNames = outputNames;
         }