c++ decoupling

Author: toudefu

third_party/iluvatar/backend/flagtree_backend_specialization/include/triton/Analysis/iluvatar_AxisInfo.h

@@ -1,6 +1,8 @@
 #ifndef ILUVATAR_TRITON_ANALYSIS_AXISINFO_H
 #define ILUVATAR_TRITON_ANALYSIS_AXISINFO_H
 
+#define FLAGTREE_SPEC_CorexFlag
+#define FLAGTREE_SPEC_AxisInfo_getCorexFlag
 #define FLAGTREE_SPEC_AxisInfo_initPessimisticStateFromFunc_ARG AxisInfo::DimVectorT *
 
 #endif

third_party/iluvatar/include/triton/Analysis/AxisInfo.h

@@ -13,8 +13,6 @@
 #include <optional>
 #include <type_traits>
 
-#define FLAGTREE_SPEC_CorexFlag
-#define FLAGTREE_SPEC_AxisInfo_getCorexFlag
 #include "flagtree_spec.h"
 
 namespace mlir::triton {

third_party/iluvatar/lib/Analysis/AxisInfo.cpp

@@ -1305,7 +1305,21 @@ void AxisInfo::initPessimisticStateFromFunc(int argNumber, T funcOp,
                                             DimVectorT *contiguity,
                                             DimVectorT *divisibility,
                                             DimVectorT *constancy) {
-  // TODO: 原先的实现
+  // liast of attributes that we care about
+  SmallVector<std::pair<DimVectorT *, std::string>> retVecs;
+  retVecs.push_back({contiguity, "tt.contiguity"});
+  retVecs.push_back({divisibility, "tt.divisibility"});
+  retVecs.push_back({constancy, "tt.constancy"});
+  // initialize attributes one by one
+  for (auto [vec, attrName] : retVecs) {
+    Attribute attr = funcOp.getArgAttr(argNumber, attrName);
+    if (auto int_attr = dyn_cast_or_null<IntegerAttr>(attr))
+      *vec = DimVectorT(contiguity->size(), int_attr.getValue().getZExtValue());
+    if (auto dense_attr = dyn_cast_or_null<DenseElementsAttr>(attr)) {
+      auto vals = dense_attr.getValues<int>();
+      *vec = DimVectorT(vals.begin(), vals.end());
+    }
+  }
 }
 #endif
 

third_party/iluvatar/lib/Analysis/CMakeLists.txt

@@ -8,7 +8,6 @@ add_triton_library(TritonAnalysis
   DEPENDS
   TritonTableGen
   TritonGPUAttrDefsIncGen
-  FlagTree_${FLAGTREE_BACKEND}_TritonAnalysis
 
   LINK_LIBS PUBLIC
   MLIRAnalysis