[codegen] renamed axis_info -> alignment_info

include/triton/codegen/alignment_info.h

@@ -1,5 +1,5 @@
-#ifndef TDL_INCLUDE_CODEGEN_AXIS_INFO_PASS_H
-#define TDL_INCLUDE_CODEGEN_AXIS_INFO_PASS_H
+#ifndef TDL_INCLUDE_CODEGEN_ALIGNMENT_INFO_PASS_H
+#define TDL_INCLUDE_CODEGEN_ALIGNMENT_INFO_PASS_H
 
 #include <set>
 #include <map>
@@ -13,7 +13,7 @@ namespace ir {
 
 namespace codegen{
 
-class axis_info {
+class alignment_info {
 private:
   // helpers
   bool is_first_axis_unit(ir::value *v);

include/triton/codegen/selection.h

@@ -25,7 +25,7 @@ class shmem_allocation;
 class tune;
 class shmem_info;
 class target;
-class axis_info;
+class alignment_info;
 
 typedef std::vector<llvm::Value*> indices_t;
 
@@ -144,7 +144,7 @@ private:
   void lower_tile_instruction(ir::instruction *src, llvm::IRBuilder<> &builder);
 
 public:
-  selection(shmem_allocation *alloc, tune *params, shmem_info *buffer_info, axis_info *ax_info, target *tgt)
+  selection(shmem_allocation *alloc, tune *params, shmem_info *buffer_info, alignment_info *ax_info, target *tgt)
     : alloc_(alloc), params_(params), buffer_info_(buffer_info), axis_info_(ax_info), tgt_(tgt){ }
 
   void run(ir::module &src, llvm::Module &dst);
@@ -158,7 +158,7 @@ private:
   tune *params_;
   target *tgt_;
   shmem_info *buffer_info_;
-  axis_info *axis_info_;
+  alignment_info *axis_info_;
   std::map<unsigned, distributed_axis> axes_;
   llvm::Value *sh_mem_ptr_;
   llvm::Value *offset_a_i_, *offset_a_k_;

include/triton/runtime/jit.h

@@ -17,7 +17,7 @@
 #include "triton/codegen/shmem_liveness.h"
 #include "triton/codegen/shmem_info.h"
 #include "triton/codegen/shmem_barriers.h"
-#include "triton/codegen/axis_info.h"
+#include "triton/codegen/alignment_info.h"
 #include "triton/codegen/target.h"
 #include "triton/codegen/vectorize.h"
 #include <functional>
@@ -91,7 +91,7 @@ public:
     codegen::optimize_dot optimize_dot;
     codegen::optimize_cse optimize_cse;
     codegen::optimize_trans optimize_trans;
-    codegen::axis_info axis_info;
+    codegen::alignment_info axis_info;
     codegen::target* target_;
   };
 

lib/codegen/alignment_info.cpp

@@ -1,4 +1,4 @@
-#include "triton/codegen/axis_info.h"
+#include "triton/codegen/alignment_info.h"
 #include "triton/ir/module.h"
 #include "triton/ir/function.h"
 #include "triton/ir/basic_block.h"
@@ -15,14 +15,14 @@ inline T add_to_cache(ir::value *i, T value, std::map<ir::value*, T> &map) {
 }
 
 
-bool axis_info::is_first_axis_unit(ir::value *x){
+bool alignment_info::is_first_axis_unit(ir::value *x){
   if(x->get_type()->is_tile_ty())
     return x->get_type()->get_tile_shapes()[0]->get_value() == 1;
   else
     return true;
 }
 
-bool axis_info::populate_is_constant(ir::value *v) {
+bool alignment_info::populate_is_constant(ir::value *v) {
   if(is_constant_.find(v) != is_constant_.end())
     return is_constant_.at(v);
   // helper for the cache
@@ -61,7 +61,7 @@ bool axis_info::populate_is_constant(ir::value *v) {
   return cache(true);
 }
 
-unsigned axis_info::populate_max_contiguous(ir::value *v){
+unsigned alignment_info::populate_max_contiguous(ir::value *v){
   if(max_contiguous_.find(v) != max_contiguous_.end())
     return max_contiguous_.at(v);
   // helper for the cache
@@ -128,7 +128,7 @@ unsigned axis_info::populate_max_contiguous(ir::value *v){
   return cache(1);
 }
 
-unsigned axis_info::populate_starting_multiple(ir::value *v){
+unsigned alignment_info::populate_starting_multiple(ir::value *v){
   if(starting_multiple_.find(v) != starting_multiple_.end())
     return starting_multiple_.at(v);
   auto cache = [this,v](unsigned value){ return add_to_cache(v, value, starting_multiple_); };
@@ -201,16 +201,16 @@ unsigned axis_info::populate_starting_multiple(ir::value *v){
   return cache(result);
 }
 
-unsigned axis_info::get_starting_multiple(ir::value* v) const {
+unsigned alignment_info::get_starting_multiple(ir::value* v) const {
   return starting_multiple_.at(v);
 }
 
-unsigned axis_info::get_max_contiguous(ir::value* v) const {
+unsigned alignment_info::get_max_contiguous(ir::value* v) const {
   return max_contiguous_.at(v);
 }
 
 
-void axis_info::run(ir::module &mod) {
+void alignment_info::run(ir::module &mod) {
   // populate constant
   for(ir::function *fn: mod.get_function_list())
   for(ir::basic_block *block: fn->blocks())

lib/codegen/selection.cpp

@@ -2,7 +2,7 @@
 #include "triton/codegen/tune.h"
 #include "triton/codegen/shmem_allocation.h"
 #include "triton/codegen/target.h"
-#include "triton/codegen/axis_info.h"
+#include "triton/codegen/alignment_info.h"
 #include "llvm/IR/InstrTypes.h"
 #include "llvm/IR/Module.h"
 #include "llvm/IR/IRBuilder.h"