Differential D130916 Diff 449049 mlir/unittests/IR/OperationSupportTest.cpp

Changeset View

Standalone View

	Show First 20 Lines • Show All 275 Lines • ▼ Show 20 Lines
	TEST(OperandStorageTest, PopulateDefaultAttrs) {			TEST(OperandStorageTest, PopulateDefaultAttrs) {
	MLIRContext context;			MLIRContext context;
	context.getOrLoadDialect<test::TestDialect>();			context.getOrLoadDialect<test::TestDialect>();
	Builder builder(&context);			Builder builder(&context);

	OpBuilder b(&context);			OpBuilder b(&context);
	auto req1 = b.getI32IntegerAttr(10);			auto req1 = b.getI32IntegerAttr(10);
	auto req2 = b.getI32IntegerAttr(60);			auto req2 = b.getI32IntegerAttr(60);
				// Verify default attributes populated post op creation.
	Operation *op = b.create<test::OpAttrMatch1>(b.getUnknownLoc(), req1, nullptr,			Operation *op = b.create<test::OpAttrMatch1>(b.getUnknownLoc(), req1, nullptr,
	nullptr, req2);			nullptr, req2);
	EXPECT_EQ(op->getAttr("default_valued_attr"), nullptr);
	op->populateDefaultAttrs();
	auto opt = op->getAttr("default_valued_attr");			auto opt = op->getAttr("default_valued_attr");
	EXPECT_NE(opt, nullptr) << *op;			EXPECT_NE(opt, nullptr) << *op;

	op->destroy();			op->destroy();
	}			}
	} // namespace			} // namespace