[mlir1684]support GaToSlice

Change-Id: I25745848993d90ad23fa1d08da55480dda91e24c
sophgo · May 19, 2023 · 2fc9217 · 2fc9217
1 parent bfdbbac
commit 2fc9217
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lib/Dialect/Tpu/Interfaces/BM1684/Slice.cpp b/lib/Dialect/Tpu/Interfaces/BM1684/Slice.cpp
@@ -33,8 +33,8 @@ void tpu::SliceOp::codegen_global_bm1684() {
   int *end_index = new int[MAX_SHAPE_DIMS];
   int *stride = new int[MAX_SHAPE_DIMS];
   // assign param and call func
-  module::getGlobalShape(input, input_shape);
   for (int i = 0; i < shape_dim; ++i) {
+    input_shape[i] = p.is_4[i];
     begin_index[i] = p.offset_4[i];
     end_index[i] = p.os_4[i] * p.step_4[i] + p.offset_4[i];
     stride[i] = p.step_4[i];

diff --git a/python/test/test_onnx.py b/python/test/test_onnx.py
@@ -213,7 +213,7 @@ def __init__(self,
             "Depth2SpaceWithPermute": (self.test_Depth2SpaceWithPermute, Y, Y, Y, N),
             "Div2Mul":          (self.test_Div2Mul,         Y, Y, Y, Y),
             "ConvSlice":        (self.test_ConvSlice,       Y, Y, Y, N),
-            "GaToSlice":        (self.test_GaToSlice,       N, Y, Y, Y),
+            "GaToSlice":        (self.test_GaToSlice,       Y, Y, Y, Y),
             "Mul2Scale":        (self.test_Mul2Scale,       Y, Y, Y, Y),
             "MatMulTranspose":  (self.test_MatMulTranspose, N, Y, Y, Y),
             "MatMulTranspose2":  (self.test_MatMulTranspose2, N, Y, Y, Y),