Резанцева Анастасия. 3822Б1ФИ1. Лабораторная 4: MLIR. Вариант 4.

[AnastasiaRezantseva]

MaxDepthPass - это пасс, который вычисляет максимальную глубину вложенности для каждого цикла в MLIR-функциях. Пасс поддерживает различные виды циклов и учитывает вложенные условные конструкции.

Рекурсивный анализ вложенности:

• Метод computeDepth() рекурсивно обходит регионы операций

• Учитывает как вложенные циклы, так и условные конструкции

• Сохраняет результаты как атрибут функции my_loop_depths (отсортированный по убыванию)

Поддерживаемые операции:

Циклы: affine.for, scf.for, scf.while, scf.forall, spirv.LoopOp

Условия: affine.if, scf.if, spirv.SelectionOp

[m-ly4]

Suggested change

	unsigned maxDepth = currentDepth;
	if (isNestingOp(op)) {
	maxDepth = std::max(maxDepth, currentDepth + 1);
	if (isNestingOp(op))
	currentDepth++;
	unsigned maxDepth = currentDepth;

[m-ly4]

WalkResult::skip() result will break recursive visiting nested operations of current op and goes to the next op at current or upper level.

Suggested change

	if (isLoopOp(op)) {
	Operation *parent = op->getParentOp();
	bool isNestedInLoop = false;
	while (parent && !isa<func::FuncOp>(parent)) {
	if (isLoopOp(parent)) {
	isNestedInLoop = true;
	break;
	}
	parent = parent->getParentOp();
	}
	if (!isNestedInLoop) {
	unsigned depth = computeDepth(op);
	if (depth > 0) {
	loopDepths.push_back(depth);
	}
	}
	}
	if (isLoopOp(op)) {
	unsigned depth = computeDepth(op);
	if (depth > 0) {
	loopDepths.push_back(depth);
	}
	return WalkResult::skip();
	}

[m-ly4]

Suggested change

	maxDepth = std::max(
	maxDepth, computeDepth(&nestedOp,
	currentDepth + (isNestingOp(op) ? 1 : 0)));
	maxDepth = std::max(maxDepth, computeDepth(&nestedOp, currentDepth));

[m-ly4]

Suggested change

	return "It's a pass, that counts the max depth of region nests in each "
	"loop.";
	return "The pass counts the max depth of control flow operations nests in each "
	"loop.";