LLZKEnforceNoOverwritePass.cpp

Go to the documentation of this file.

//===-- LLZKEnforceNoOverwritePass.cpp --------------------------*- C++ -*-===//
//
// Part of the LLZK Project, under the Apache License v2.0.
// See LICENSE.txt for license information.
// Copyright 2026 Project LLZK
// SPDX-License-Identifier: Apache-2.0
//
//===----------------------------------------------------------------------===//
//===----------------------------------------------------------------------===//
 
#include "llzk/Analysis/AnalysisUtil.h"
#include "llzk/Analysis/MemberOverwriteAnalysis.h"
#include "llzk/Dialect/Function/IR/Ops.h"
#include "llzk/Dialect/Struct/IR/Ops.h"
#include "llzk/Transforms/LLZKTransformationPasses.h"
 
#include <mlir/Analysis/DataFlowFramework.h>
 
#include <llvm/Support/Debug.h>
 
#include <memory>
 
namespace llzk {
#define GEN_PASS_DECL_ENFORCENOMEMBEROVERWRITEPASS
#define GEN_PASS_DEF_ENFORCENOMEMBEROVERWRITEPASS
#include "llzk/Transforms/LLZKTransformationPasses.h.inc"
} // namespace llzk
 
#define DEBUG_TYPE "llzk-enforce-no-overwrites-pass"
 
using std::make_unique;
 
using namespace mlir;
 
namespace llzk {
 
using namespace function;
using namespace component;
 
class EnforceNoMemberOverwritePass
    : public llzk::impl::EnforceNoMemberOverwritePassBase<EnforceNoMemberOverwritePass> {
 
  void runOnOperation() override {
    getOperation()->walk([this](StructDefOp structDef) {
      auto result = analyzeStruct(structDef);
      if (failed(result)) {
        signalPassFailure();
      }
      const auto &[overwrites, written] = *result;
 
      for (auto member : structDef.getMemberDefs()) {
        if (!written.contains(member.getSymName())) {
          signalPassFailure();
        }
      }
 
      if (!overwrites.empty()) {
        signalPassFailure();
      }
    });
  }
};
 
std::unique_ptr<mlir::Pass> createNoOverwritesPass() {
  return make_unique<EnforceNoMemberOverwritePass>();
}
} // namespace llzk