Mamaeva olga lab4 var3

mlir/compiler-course/CMakeLists.txt

@@ -1,3 +1,16 @@
-list_subdirs(subdirs ${CMAKE_CURRENT_SOURCE_DIR})
-add_subdirs(subdirs "COMPILER COURSE (MLIR)")
-set(MLIR_TEST_DEPENDS ${MLIR_TEST_DEPENDS} PARENT_SCOPE)
+set(Title "RemPass")
+set(Student "Mamaeva_Olga")
+set(Group "FIIT3")
+set(TARGET_NAME "${Title}_${Student}_${Group}_MLIR")
+
+file(GLOB_RECURSE SOURCES *.cpp *.h *.hpp)
+
+add_llvm_pass_plugin(${TARGET_NAME}
+    ${SOURCES}
+    DEPENDS
+    intrinsics_gen
+    MLIRBuiltinLocationAttributesIncGen
+    BUILDTREE_ONLY
+)
+
+set(MLIR_TEST_DEPENDS ${TARGET_NAME} ${MLIR_TEST_DEPENDS} PARENT_SCOPE)

mlir/compiler-course/rem_pass.cpp

@@ -0,0 +1,66 @@
+#include "mlir/Dialect/Arith/IR/Arith.h"
+#include "mlir/Dialect/Func/IR/FuncOps.h"
+#include "mlir/IR/PatternMatch.h"
+#include "mlir/Pass/Pass.h"
+#include "mlir/Tools/Plugins/PassPlugin.h"
+
+using namespace mlir;
+
+namespace {
+class RemPass_Mamaeva_Olga_FIIT3_MLIR
+    : public PassWrapper<RemPass_Mamaeva_Olga_FIIT3_MLIR,
+                         OperationPass<ModuleOp>> {
+public:
+  StringRef getArgument() const final {
+    return "rem-pass-mamaeva-olga-fiit3-mlir";
+  }
+
+  StringRef getDescription() const final {
+    return "Replace remainder operations with div+mul+sub sequence";
+  }
+
+  void runOnOperation() override {
+    ModuleOp module = getOperation();
+    OpBuilder builder(module);
+
+    module.walk([&](arith::RemSIOp remOp) {
+      expandRemainder<arith::DivSIOp>(remOp, builder);
+    });
+
+    module.walk([&](arith::RemUIOp remOp) {
+      expandRemainder<arith::DivUIOp>(remOp, builder);
+    });
+  }
+
+private:
+  template <typename DivOp>
+  void expandRemainder(Operation *op, OpBuilder &builder) {
+    Value lhs = op->getOperand(0);
+    Value rhs = op->getOperand(1);
+    Location loc = op->getLoc();
+
+    builder.setInsertionPoint(op);
+
+    Value div = builder.create<DivOp>(loc, lhs, rhs);
+    Value mul = builder.create<arith::MulIOp>(loc, div, rhs);
+    Value sub = builder.create<arith::SubIOp>(loc, lhs, mul);
+
+    op->getResult(0).replaceAllUsesWith(sub);
+    op->erase();
+  }
+};
+} // namespace
+
+MLIR_DECLARE_EXPLICIT_TYPE_ID(RemPass_Mamaeva_Olga_FIIT3_MLIR)
+MLIR_DEFINE_EXPLICIT_TYPE_ID(RemPass_Mamaeva_Olga_FIIT3_MLIR)
+
+static mlir::PassPluginLibraryInfo
+getRemPass_Mamaeva_Olga_FIIT3_MLIRPluginInfo() {
+  return {MLIR_PLUGIN_API_VERSION, "RemPass_Mamaeva_Olga_FIIT3_MLIR", "1.0",
+          []() { mlir::PassRegistration<RemPass_Mamaeva_Olga_FIIT3_MLIR>(); }};
+}
+
+extern "C" LLVM_ATTRIBUTE_WEAK mlir::PassPluginLibraryInfo
+mlirGetPassPluginInfo() {
+  return getRemPass_Mamaeva_Olga_FIIT3_MLIRPluginInfo();
+}

mlir/test/compiler-course/test.mlir

@@ -0,0 +1,85 @@
+// RUN: mlir-opt -load-pass-plugin=%mlir_lib_dir/RemPass_Mamaeva_Olga_FIIT3_MLIR%shlibext \
+// RUN: --pass-pipeline="builtin.module(RemPass_Mamaeva_Olga_FIIT3_MLIR)" %s | FileCheck %s
+
+// ==================================================
+// Тест для знакового остатка (remsi)
+// ==================================================
+// CHECK-LABEL: func.func @test_remsi
+func.func @test_remsi(%arg0: i32, %arg1: i32) -> i32 {
+  // Проверяем что остаток заменяется на: a - (a / b) * b
+  // CHECK: %[[DIV:.*]] = arith.divsi %arg1, %arg0 : i32
+  // CHECK: %[[MUL:.*]] = arith.muli %[[DIV]], %arg0 : i32
+  // CHECK: %[[RES:.*]] = arith.subi %arg1, %[[MUL]] : i32
+  %rem = arith.remsi %arg1, %arg0 : i32
+
+  // CHECK: return %[[RES]]
+  return %rem : i32
+}
+
+// ==================================================
+// Тест для беззнакового остатка (remui)
+// ==================================================
+// CHECK-LABEL: func.func @test_remui
+func.func @test_remui(%arg0: i32, %arg1: i32) -> i32 {
+  // Для беззнакового используем divui
+  // CHECK: %[[DIV:.*]] = arith.divui %arg1, %arg0 : i32
+  // CHECK: %[[MUL:.*]] = arith.muli %[[DIV]], %arg0 : i32
+  // CHECK: %[[RES:.*]] = arith.subi %arg1, %[[MUL]] : i32
+  %rem = arith.remui %arg1, %arg0 : i32
+
+  // CHECK: return %[[RES]]
+  return %rem : i32
+}
+
+// ==================================================
+// Тест для 64-битных целых
+// ==================================================
+// CHECK-LABEL: func.func @test_i64
+func.func @test_i64(%arg0: i64, %arg1: i64) -> i64 {
+  // Проверяем работу с другим типом данных
+  // CHECK: arith.divsi
+  // CHECK: arith.muli
+  // CHECK: arith.subi
+  %rem = arith.remsi %arg1, %arg0 : i64
+
+  // CHECK: return
+  return %rem : i64
+}
+
+// ==================================================
+// Тест с несколькими операциями остатка
+// ==================================================
+// CHECK-LABEL: func.func @test_multiple
+func.func @test_multiple(%arg0: i32, %arg1: i32) -> (i32, i32) {
+  // Проверяем замену нескольких операций в одной функции
+  // CHECK: arith.divsi
+  // CHECK: arith.muli
+  // CHECK: arith.subi
+  %rem1 = arith.remsi %arg1, %arg0 : i32
+
+  // CHECK: arith.divsi
+  // CHECK: arith.muli
+  // CHECK: arith.subi
+  %rem2 = arith.remsi %arg0, %arg1 : i32
+
+  // CHECK: return
+  return %rem1, %rem2 : i32, i32
+}
+
+// ==================================================
+// Тест с константными значениями
+// ==================================================
+// CHECK-LABEL: func.func @test_constants
+func.func @test_constants() -> i32 {
+  %c42 = arith.constant 42 : i32
+  %c5 = arith.constant 5 : i32
+
+  // Проверяем работу с константами
+  // CHECK: arith.divsi
+  // CHECK: arith.muli
+  // CHECK: arith.subi
+  %rem = arith.remsi %c42, %c5 : i32
+
+  // CHECK: return
+  return %rem : i32
+}