fix bug

PaddlePaddle · Aug 24, 2022 · cce1792 · cce1792
1 parent 3b4e976
commit cce1792
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cinn/hlir/op/contrib/scatter.cc b/cinn/hlir/op/contrib/scatter.cc
@@ -85,7 +85,7 @@ ir::Tensor Scatter(const ir::Tensor &A,
             offset = offset * C->shape[i] + indices[i];
           }
         }
-        offset   = common::AutoSimplify(offset);
+//        offset   = common::AutoSimplify(offset);
         auto idx = lang::CallExtern(extern_fun_name, {new_B, new_B->shape[-1], Expr(0), offset, Expr(1)});
         //        auto idx = lang::CallExtern(extern_fun_name, {new_B, new_B->shape[-1], indices[pos_axis], offset,
         //        Expr(1)});