Better textual representation

2022-04-07 20:44:41 +08:00
parent 0864b253bb
commit 6002340456
3 changed files with 10 additions and 10 deletions
--- a/python/src/triton.cc
+++ b/python/src/triton.cc
@@ -1192,7 +1192,7 @@ void init_triton_ir(py::module &&m) {
        auto loc = self.getUnknownLoc();
        auto argType = arg.getType().dyn_cast<mlir::RankedTensorType>().getElementType();
        return self.create<mlir::triton::ReshapeOp>(
-          loc, mlir::RankedTensorType::get(shape, argType), arg, self.getI64ArrayAttr(shape)
+          loc, mlir::RankedTensorType::get(shape, argType), arg
        );
      })
      .def("create_cat", [](mlir::OpBuilder &self, mlir::Value &lhs, mlir::Value &rhs) -> mlir::Value {