[Refactor] Introduce NeuraAttributes.h to manage attribute constants

include/NeuraDialect/Mapping/HeuristicMapping/HeuristicMapping.h

@@ -3,6 +3,7 @@
 
 #include "NeuraDialect/Mapping/Mapping.h"
 #include "NeuraDialect/Mapping/MappingState.h"
+#include "NeuraDialect/NeuraAttributes.h"
 #include <climits>
 #include <set>
 
@@ -21,14 +22,14 @@ class HeuristicMapping : public Mapping {
 
   std::string getName() const override {
     if (max_location_to_try == 1 && max_backtrack_depth == 1) {
-      return "simple";
+      return attr::val::kSimple.str();
     } else if (max_location_to_try == INT_MAX && max_backtrack_depth == 1) {
-      return "greedy";
+      return attr::val::kGreedy.str();
     } else if (max_location_to_try == INT_MAX &&
                max_backtrack_depth == INT_MAX) {
-      return "exhaustive";
+      return attr::val::kExhaustive.str();
     } else {
-      return "customized";
+      return attr::val::kCustomized.str();
     }
   }
 

include/NeuraDialect/NeuraAttributes.h

@@ -0,0 +1,50 @@
+#pragma once
+
+#include "llvm/ADT/StringRef.h"
+
+namespace mlir {
+namespace neura {
+
+namespace attr {
+
+// Attribute Keys
+constexpr llvm::StringLiteral kDataflowMode = "dataflow_mode";
+constexpr llvm::StringLiteral kMappingMode = "mapping_mode";
+constexpr llvm::StringLiteral kMappingStrategy = "mapping_strategy";
+constexpr llvm::StringLiteral kBacktrackConfig = "backtrack_config";
+constexpr llvm::StringLiteral kDumpMappingTable = "dump_mapping_table";
+
+// Identification & Results
+constexpr llvm::StringLiteral kDfgId = "dfg_id";
+constexpr llvm::StringLiteral kMappingInfo = "mapping_info";
+constexpr llvm::StringLiteral kXTiles = "x_tiles";
+constexpr llvm::StringLiteral kYTiles = "y_tiles";
+constexpr llvm::StringLiteral kCompiledII = "compiled_ii";
+constexpr llvm::StringLiteral kRecMII = "rec_mii";
+constexpr llvm::StringLiteral kResMII = "res_mii";
+
+// Attribute Values & Constants
+namespace val {
+// Strategy & Mode
+constexpr llvm::StringLiteral kSpatialOnly = "spatial-only";
+constexpr llvm::StringLiteral kSpatialTemporal = "spatial-temporal";
+constexpr llvm::StringLiteral kTemporal = "temporal";
+constexpr llvm::StringLiteral kHeuristic = "heuristic";
+constexpr llvm::StringLiteral kCustomized = "customized";
+constexpr llvm::StringLiteral kSimple = "simple";
+constexpr llvm::StringLiteral kGreedy = "greedy";
+constexpr llvm::StringLiteral kExhaustive = "exhaustive";
+
+// Identifiers
+constexpr llvm::StringLiteral kModeSteering = "steering";
+constexpr llvm::StringLiteral kModePredicate = "predicate";
+
+// Operation Logic
+constexpr llvm::StringLiteral kOpFused = "fused_op";
+constexpr llvm::StringLiteral kNeuraFusedOp = "neura.fused_op";
+
+} // namespace val
+
+} // namespace attr
+} // namespace neura
+} // namespace mlir

lib/NeuraDialect/Transforms/CanonicalizeCastPass.cpp

@@ -1,3 +1,4 @@
+#include "Common/AcceleratorAttrs.h"
 #include "NeuraDialect/NeuraOps.h"
 #include "mlir/Dialect/Func/IR/FuncOps.h"
 #include "mlir/Dialect/LLVMIR/LLVMDialect.h"
@@ -113,14 +114,16 @@ struct CanonicalizeCastPass
     module_op.walk([&](Operation *op) {
       Region *region = nullptr;
       if (auto func_op = dyn_cast<func::FuncOp>(op)) {
-        auto accel_attr = func_op->getAttrOfType<StringAttr>("accelerator");
-        if (!accel_attr || accel_attr.getValue() != "neura") {
+        auto accel_attr =
+            func_op->getAttrOfType<StringAttr>(accel::kAcceleratorAttr);
+        if (!accel_attr || accel_attr.getValue() != accel::kNeuraTarget) {
           return;
         }
         region = &func_op.getBody();
       } else if (auto llvm_func = dyn_cast<LLVM::LLVMFuncOp>(op)) {
-        auto accel_attr = llvm_func->getAttrOfType<StringAttr>("accelerator");
-        if (!accel_attr || accel_attr.getValue() != "neura") {
+        auto accel_attr =
+            llvm_func->getAttrOfType<StringAttr>(accel::kAcceleratorAttr);
+        if (!accel_attr || accel_attr.getValue() != accel::kNeuraTarget) {
           return;
         }
         region = &llvm_func.getBody();

lib/NeuraDialect/Transforms/CanonicalizeLiveInPass.cpp

@@ -1,3 +1,4 @@
+#include "Common/AcceleratorAttrs.h"
 #include "NeuraDialect/NeuraDialect.h"
 #include "NeuraDialect/NeuraOps.h"
 #include "mlir/Dialect/Func/IR/FuncOps.h"
@@ -783,14 +784,16 @@ struct CanonicalizeLiveInPass
     module_op.walk([&](Operation *op) {
       Region *region = nullptr;
       if (auto func_op = dyn_cast<func::FuncOp>(op)) {
-        auto accel_attr = func_op->getAttrOfType<StringAttr>("accelerator");
-        if (!accel_attr || accel_attr.getValue() != "neura") {
+        auto accel_attr =
+            func_op->getAttrOfType<StringAttr>(accel::kAcceleratorAttr);
+        if (!accel_attr || accel_attr.getValue() != accel::kNeuraTarget) {
           return;
         }
         region = &func_op.getBody();
       } else if (auto llvm_func = dyn_cast<LLVM::LLVMFuncOp>(op)) {
-        auto accel_attr = llvm_func->getAttrOfType<StringAttr>("accelerator");
-        if (!accel_attr || accel_attr.getValue() != "neura") {
+        auto accel_attr =
+            llvm_func->getAttrOfType<StringAttr>(accel::kAcceleratorAttr);
+        if (!accel_attr || accel_attr.getValue() != accel::kNeuraTarget) {
           return;
         }
         region = &llvm_func.getBody();

lib/NeuraDialect/Transforms/CanonicalizeReturnPass.cpp

@@ -1,3 +1,4 @@
+#include "Common/AcceleratorAttrs.h"
 #include "NeuraDialect/NeuraDialect.h"
 #include "NeuraDialect/NeuraOps.h"
 #include "NeuraDialect/NeuraPasses.h"
@@ -191,7 +192,8 @@ struct CanonicalizeReturnPass
   void runOnOperation() override {
     func::FuncOp func_op = getOperation();
     // Checks for neura accelerator attribute.
-    auto accel_attr = func_op->getAttrOfType<StringAttr>("accelerator");
+    auto accel_attr =
+        func_op->getAttrOfType<StringAttr>(accel::kAcceleratorAttr);
     if (!accel_attr) {
       return;
     }

lib/NeuraDialect/Transforms/GenerateCodePass.cpp

@@ -1,3 +1,4 @@
+#include "Common/AcceleratorAttrs.h"
 #include "mlir/Pass/Pass.h"
 #include "mlir/IR/BuiltinOps.h"
 #include "mlir/IR/Operation.h"
@@ -23,6 +24,7 @@
 
 #include "NeuraDialect/Architecture/Architecture.h"
 #include "NeuraDialect/NeuraOps.h"
+#include "NeuraDialect/NeuraAttributes.h"
 
 using namespace mlir;
 using namespace neura;
@@ -410,16 +412,16 @@ struct GenerateCodePass
 
   std::pair<int, int> getArrayDimensions(func::FuncOp function) {
     int columns = 4, rows = 4; // default 4x4 CGRA.
-    if (auto mapping_info = function->getAttrOfType<DictionaryAttr>("mapping_info")) {
-      if (auto x_tiles = dyn_cast_or_null<IntegerAttr>(mapping_info.get("x_tiles"))) columns = x_tiles.getInt();
-      if (auto y_tiles = dyn_cast_or_null<IntegerAttr>(mapping_info.get("y_tiles"))) rows   = y_tiles.getInt();
+    if (auto mapping_info = function->getAttrOfType<DictionaryAttr>(attr::kMappingInfo)) {
+      if (auto x_tiles = dyn_cast_or_null<IntegerAttr>(mapping_info.get(attr::kXTiles))) columns = x_tiles.getInt();
+      if (auto y_tiles = dyn_cast_or_null<IntegerAttr>(mapping_info.get(attr::kYTiles))) rows   = y_tiles.getInt();
     }
     return {columns, rows};
   }
 
   int getCompiledII(func::FuncOp function) {
-    if (auto mapping_info = function->getAttrOfType<DictionaryAttr>("mapping_info")) {
-      if (auto compiled_ii = dyn_cast_or_null<IntegerAttr>(mapping_info.get("compiled_ii"))) {
+    if (auto mapping_info = function->getAttrOfType<DictionaryAttr>(attr::kMappingInfo)) {
+      if (auto compiled_ii = dyn_cast_or_null<IntegerAttr>(mapping_info.get(attr::kCompiledII))) {
         return compiled_ii.getInt();
       }
     }
@@ -933,7 +935,7 @@ struct GenerateCodePass
 
   // Helper to extract dfg_id from operation.
   static int getDfgId(Operation *op) {
-    if (auto id_attr = op->getAttrOfType<IntegerAttr>("dfg_id")) {
+    if (auto id_attr = op->getAttrOfType<IntegerAttr>(attr::kDfgId)) {
       return id_attr.getInt();
     }
     return -1;
@@ -1664,8 +1666,8 @@ struct GenerateCodePass
     ModuleOp module = getOperation();
 
     for (auto func : module.getOps<func::FuncOp>()) {
-      auto accel = func->getAttrOfType<StringAttr>("accelerator");
-      if (!accel || accel.getValue() != "neura") continue;
+      auto accel = func->getAttrOfType<StringAttr>(accel::kAcceleratorAttr);
+      if (!accel || accel.getValue() != accel::kNeuraTarget) continue;
 
       auto [columns, rows] = getArrayDimensions(func);
       Topology topo = getTopologyFromArchitecture(columns, rows);

lib/NeuraDialect/Transforms/GraphMining/GraMi.cpp

@@ -1,3 +1,5 @@
+#include "Common/AcceleratorAttrs.h"
+#include "NeuraDialect/NeuraAttributes.h"
 #include "NeuraDialect/Transforms/GraphMining/GraMi.h"
 #include "NeuraDialect/Mapping/mapping_util.h"
 #include "mlir/Dialect/Func/IR/FuncOps.h"
@@ -204,13 +206,13 @@ std::vector<PatternWithSelectedInstances> GraMi::mineFrequentSubgraphs() {
   auto derive_label = [](mlir::Operation* op, const std::string& fallback_label) -> std::string {
     if (!op) return fallback_label;
     auto name = op->getName().getStringRef();
-    if (name.ends_with("fused_op") || name.contains("neura.fused_op")) {
+    if (name.ends_with(attr::val::kOpFused) || name.contains(attr::val::kNeuraFusedOp)) {
       if (auto attr = op->getAttr("pattern_name")) {
         if (auto str_attr = mlir::dyn_cast<mlir::StringAttr>(attr)) {
           return std::string("fused_op:") + str_attr.getValue().str();
         }
       }
-      return std::string("fused_op");
+      return std::string(attr::val::kOpFused);
     }
     return fallback_label;
   };

lib/NeuraDialect/Transforms/InsertCtrlMovPass.cpp

@@ -1,3 +1,4 @@
+#include "Common/AcceleratorAttrs.h"
 #include "NeuraDialect/NeuraDialect.h"
 #include "NeuraDialect/NeuraOps.h"
 #include "NeuraDialect/NeuraPasses.h"
@@ -20,7 +21,7 @@ struct InsertCtrlMovForNeuraOps : public RewritePattern {
 
   LogicalResult matchAndRewrite(Operation *op,
                                 PatternRewriter &rewriter) const override {
-    if (op->getDialect()->getNamespace() != "neura" ||
+    if (op->getDialect()->getNamespace() != accel::kNeuraTarget ||
         isa<neura::CtrlMovOp>(op)) {
       return failure();
     }

lib/NeuraDialect/Transforms/InsertDataMovPass.cpp

@@ -1,3 +1,4 @@
+#include "Common/AcceleratorAttrs.h"
 #include "NeuraDialect/NeuraDialect.h"
 #include "NeuraDialect/NeuraOps.h"
 #include "NeuraDialect/NeuraPasses.h"
@@ -22,7 +23,7 @@ struct InsertDataMovForNeuraOps : public RewritePattern {
 
   LogicalResult matchAndRewrite(Operation *op,
                                 PatternRewriter &rewriter) const override {
-    if (op->getDialect()->getNamespace() != "neura" ||
+    if (op->getDialect()->getNamespace() != accel::kNeuraTarget ||
         isa<neura::DataMovOp>(op)) {
       return failure();
     }

lib/NeuraDialect/Transforms/LeveragePredicatedValuePass.cpp

@@ -1,3 +1,4 @@
+#include "Common/AcceleratorAttrs.h"
 #include "NeuraDialect/NeuraDialect.h"
 #include "NeuraDialect/NeuraOps.h"
 #include "NeuraDialect/NeuraPasses.h"
@@ -33,8 +34,9 @@ struct LeveragePredicatedValuePass
 
     // Processes each function.
     module.walk([&](FunctionOpInterface func) {
-      auto accel_attr = func->getAttrOfType<StringAttr>("accelerator");
-      if (!accel_attr || accel_attr.getValue() != "neura") {
+      auto accel_attr =
+          func->getAttrOfType<StringAttr>(accel::kAcceleratorAttr);
+      if (!accel_attr || accel_attr.getValue() != accel::kNeuraTarget) {
         return;
       }
       // Converts block argument types to predicated values.
@@ -107,7 +109,7 @@ struct LeveragePredicatedValuePass
   // Converts a single operation to use predicated values.
   LogicalResult applyPredicatedDataType(Operation *op) {
     // Skips if not a Neura op.
-    if (op->getDialect()->getNamespace() != "neura") {
+    if (op->getDialect()->getNamespace() != accel::kNeuraTarget) {
       return success();
     }