[ssbuffer](feat) extend scenario of merge cube block

Signed-off-by: fishofnanqi <1074959344@qq.com>

Author: zhuxinguang33

third_party/ascend/include/DynamicCVPipeline/ComputeBlockOpt/MergeCubeBlockPass.h

@@ -99,6 +99,7 @@ class MergeCubeBlockPass
                               BlockDependencyGraph &graph);
   bool checkSameSourceAndSink(int blockId1, int blockId2,
                               BlockDependencyGraph &graph);
+  bool hasSameDepth(int blockId1, int blockId2, BlockDependencyGraph &graph);
   bool checkNoCycle(int blockId1, int blockId2, BlockDependencyGraph &graph,
                     const MemoryDependenceGraph &memGraph,
                     ComputeBlockIdManager &bm);

third_party/ascend/lib/DynamicCVPipeline/ComputeBlockOpt/MergeCubeBlock.cpp

@@ -303,21 +303,28 @@ bool MergeCubeBlockPass::canMergeBlocks(
   }
 
   // Step 2: Check if they have same source and sink with no other nodes
-  if (!checkSameSourceAndSink(blockId1, blockId2, graph)) {
+  if (checkSameSourceAndSink(blockId1, blockId2, graph)) {
     LDBG("Blocks " << blockId1 << " and " << blockId2
-                   << " cannot merge: different source or sink\n");
-    return false;
+                   << " can merge: same source or sink\n");
+    return true;
   }
 
   // Step 3: Check if merging would create a cycle
-  if (checkNoCycle(blockId1, blockId2, graph, memGraph, bm)) {
+  if (!checkNoCycle(blockId1, blockId2, graph, memGraph, bm)) {
     LDBG("Blocks " << blockId1 << " and " << blockId2
-                   << " can merge: no cycle detected\n");
-    return true;
+                   << "cannot merge: would create cycle\n");
+    return false;
   }
 
+  // Step 4: cube block in the same depth
+  if (hasSameDepth(blockId1, blockId2, graph)) {
+    LDBG("Blocks " << blockId1 << " and " << blockId2
+                    << " can merge: cube blocks have same depth\n");
+    return true;
+    }
+
   LDBG("Blocks " << blockId1 << " and " << blockId2
-                 << " cannot merge: would create cycle\n");
+                 << " cannot merge: unsupport scenario\n");
   return false;
 }
 
@@ -368,6 +375,20 @@ bool MergeCubeBlockPass::checkSameSourceAndSink(int blockId1, int blockId2,
   return true;
 }
 
+bool MergeCubeBlockPass::hasSameDepth(int blockId1, int blockId2,
+                                      BlockDependencyGraph &graph) {
+
+  // Get block nodes
+  BlockNode *node1 = graph.getBlockNode(blockId1);
+  BlockNode *node2 = graph.getBlockNode(blockId2);
+
+  if (!node1 || !node2) {
+    return false;
+  }
+
+  return node1->depth == node2->depth;
+}
+
 llvm::SmallVector<int> MergeCubeBlockPass::filterBlocksByType(
     int blockId, llvm::SmallVector<int> blocks, BlockDependencyGraph &graph) {