Address code review comments

Signed-off-by: Tiotto, Ettore <[email protected]>

Author: etiotto

third_party/intel/include/Analysis/AxisInfo.h

@@ -42,11 +42,10 @@ class ModuleAxisInfoAnalysis : public triton::ModuleAxisInfoAnalysis {
     }
   }
 
-  AxisInfo *getAxisInfo(Value value) const {
+  AxisInfo *getAxisInfo(Value value) {
     auto funcOp =
         value.getParentRegion()->getParentOfType<FunctionOpInterface>();
-    auto *axisInfoMap =
-        const_cast<ModuleAxisInfoAnalysis *>(this)->getFuncData(funcOp);
+    auto *axisInfoMap = getFuncData(funcOp);
     if (!axisInfoMap) {
       return nullptr;
     }
@@ -57,9 +56,9 @@ class ModuleAxisInfoAnalysis : public triton::ModuleAxisInfoAnalysis {
     return &(it->second);
   }
 
-  unsigned getPtrContiguity(Value ptr) const;
-  unsigned getPtrAlignment(Value ptr) const;
-  unsigned getMaskAlignment(Value mask) const;
+  unsigned getPtrContiguity(Value ptr);
+  unsigned getPtrAlignment(Value ptr);
+  unsigned getMaskAlignment(Value mask);
 
 private:
   void initialize(FunctionOpInterface funcOp);

third_party/intel/lib/Analysis/AxisInfo.cpp

@@ -1159,7 +1159,7 @@ void AxisInfoAnalysis::visitForOpInductionVar(
 
 } // anonymous namespace
 
-unsigned ModuleAxisInfoAnalysis::getPtrContiguity(Value ptr) const {
+unsigned ModuleAxisInfoAnalysis::getPtrContiguity(Value ptr) {
   auto tensorTy = ttgi::getRankedTensorType(ptr.getType());
   if (!tensorTy)
     return 1;
@@ -1181,7 +1181,7 @@ unsigned ModuleAxisInfoAnalysis::getPtrContiguity(Value ptr) const {
   return contiguity;
 }
 
-unsigned ModuleAxisInfoAnalysis::getPtrAlignment(Value ptr) const {
+unsigned ModuleAxisInfoAnalysis::getPtrAlignment(Value ptr) {
   auto tensorTy = ttgi::getRankedTensorType(ptr.getType());
   if (!tensorTy)
     return 1;
@@ -1211,7 +1211,7 @@ unsigned ModuleAxisInfoAnalysis::getPtrAlignment(Value ptr) const {
   return alignment;
 }
 
-unsigned ModuleAxisInfoAnalysis::getMaskAlignment(Value mask) const {
+unsigned ModuleAxisInfoAnalysis::getMaskAlignment(Value mask) {
   auto tensorTy = ttgi::getRankedTensorType(mask.getType());
   if (!tensorTy)
     return 1;