diff --git a/tests/mlir-tblgen-builder/test_basic.cpp b/tests/mlir-tblgen-builder/test_basic.cpp index 6b06b27..8ff7049 100644 --- a/tests/mlir-tblgen-builder/test_basic.cpp +++ b/tests/mlir-tblgen-builder/test_basic.cpp @@ -5,16 +5,16 @@ int main() { builder::Shape shape({10, 10}); auto pType = builder::PrimitiveType::F32(); - builder::Type type(shape, pType); + builder::Type OutType(shape, pType); std::vector data(100); builder::Tensor tensor(shape, data); - auto in1 = builder.CreateInput(type); - auto op1 = builder::mhlo::ConstOp::build(builder, type, tensor); - auto op2 = builder::mhlo::ConstOp::build(builder, type, tensor); - auto op3 = builder::mhlo::AddOp::build(builder, type, op1, op2); - auto op4 = builder::mhlo::MulOp::build(builder, type, op3, in1); + auto in1 = builder.CreateInput(OutType); + auto op1 = builder::mhlo::ConstOp::build(builder, OutType, tensor); + auto op2 = builder::mhlo::ConstOp::build(builder, OutType, tensor); + auto op3 = builder::mhlo::AddOp::build(builder, OutType, op1, op2); + auto op4 = builder::mhlo::MulOp::build(builder, OutType, op3, in1); builder.SetOutput(std::vector({op4})); builder.DumpModule();