Fix issues to compile Yolov3-12. (#1726)

Stop conversion to enable onnx.LeakyRelu on NNPA when the stickified layouts of input and output are not the same. Signed-off-by: Yasushi Negishi <[email protected]>
onnx · Sep 26, 2022 · 20667cd · 20667cd
1 parent df104fb
commit 20667cd
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/Accelerators/NNPA/Dialect/ZHigh/ZHighCombine.td b/src/Accelerators/NNPA/Dialect/ZHigh/ZHighCombine.td
@@ -156,7 +156,7 @@ def SigmoidLayoutChangingPattern : Pat<
 //   - %X should have static shape, and %alpha should be constant.
 //
 def ReplaceONNXLeakyReluPattern: Pat<
-  (ZHighStickOp (ONNXLeakyReluOp:$out (ZHighUnstickOp $X), $alpha),
+  (ZHighStickOp:$stickout (ONNXLeakyReluOp:$out (ZHighUnstickOp $X), $alpha),
                 $layout),
   (ZHighSubOp
      (ZHighReluOp $X, (returnType $X)),
@@ -166,7 +166,8 @@ def ReplaceONNXLeakyReluPattern: Pat<
                                             $layout),
                               (returnType $X)),
                   (returnType $X))),
-  [(IsStaticShapeTensor $X), (IsPlusConstantFloat $alpha)]
+  [(IsStaticShapeTensor $X), (IsPlusConstantFloat $alpha),
+   (SameLayout $X, $stickout)]
 >;
 
 #endif // ZHIGH_COMBINE