Differential D84171 Diff 279236 mlir/test/lib/IR/TestTypes.cpp

Changeset View

Standalone View

mlir/test/lib/IR/TestTypes.cpp

This file was added.

				//===- TestTypes.cpp - Test passes for MLIR types -------------------------===//
				//
				// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
				// See https://llvm.org/LICENSE.txt for license information.
				// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
				//
				//===----------------------------------------------------------------------===//

				#include "TestTypes.h"
				#include "TestDialect.h"
				#include "mlir/Pass/Pass.h"

				using namespace mlir;

				namespace {
				struct TestRecursiveTypesPass
				: public PassWrapper<TestRecursiveTypesPass, FunctionPass> {
				void runOnFunction() override {
				FuncOp func = getFunction();

				// Just make sure recurisve types are printed and parsed.
				if (func.getName() == "roundtrip")
				return;

				// Create a recursive type and print it as a part of a dummy op.
				if (func.getName() == "create") {
				auto type =
				rriddleUnsubmitted Done Reply Inline Actions nit: Can you move the body of this function out to a new function? rriddle: nit: Can you move the body of this function out to a new function?
				TestRecursiveType::create(&getContext(), "some_long_and_unique_name");
				if (failed(type.setBody(type))) {
				func.emitError("expected to be able to set the type body");
				return signalPassFailure();
				}

				// Setting the same body is fine.
				if (failed(type.setBody(type))) {
				func.emitError(
				"expected to be able to set the type body to the same value");
				return signalPassFailure();
				}

				// Setting a different body is not.
				if (succeeded(type.setBody(IndexType::get(&getContext())))) {
				func.emitError(
				"not expected to be able to change function body more than once");
				return signalPassFailure();
				}

				// Expecting to get the same type for the same name.
				auto other =
				TestRecursiveType::create(&getContext(), "some_long_and_unique_name");
				if (type != other) {
				func.emitError("expected type name to be the uniquing key");
				return signalPassFailure();
				}

				// Create the op to check how the type is printed.
				OperationState state(func.getLoc(), "test.dummy_type_test_op");
				state.addTypes(type);
				func.getBody().front().push_front(Operation::create(state));

				return;
				}

				// Unknown key.
				func.emitOpError() << "unexpected function name";
				signalPassFailure();
				}
				};
				} // end namespace

				namespace mlir {

				void registerTestRecursiveTypesPass() {
				PassRegistration<TestRecursiveTypesPass> reg(
				"test-recursive-types", "Test support for recursive types");
				}

				} // end namespace mlir