align_utils.hpp

#pragma once
 
#include <vector>
#include "lamppp/tensor/data_type.hpp"
 
#define LMP_MAX_DIMS 16  
 
namespace lmp::tensor::detail {
 
using stride_t = int64_t;
using stride_list = std::vector<stride_t>;
using shape_list = std::vector<size_t>;
 
class AlignUtil {
 public:
  explicit AlignUtil(const std::vector<size_t>& a_shape,
                     const std::vector<size_t>& b_shape);
 
  std::vector<size_t> aligned_shape_;
  std::vector<stride_t> aligned_stride_;
  size_t aligned_size_;
 
 private:
  static std::vector<size_t> calc_aligned_shape(const std::vector<size_t>& a_shape,
                                         const std::vector<size_t>& b_shape);
  std::vector<stride_t> calc_aligned_stride();
};
 
}  // namespace lmp::tensor::detail