Fix build failure after 91d58f5

Signed-off-by: Whitney Tsang <[email protected]>

Author: whitneywhtsang

third_party/intel/lib/Analysis/AxisInfo.cpp

@@ -1314,8 +1314,7 @@ unsigned ModuleAxisInfoAnalysis::getContiguity(Value value) {
   // FIXME: This is not as good as it could be, as we don't need to restrict
   // the analysis to one dimension. We should determine contiguity on the
   // flattenOuts() layout
-  auto linAttr =
-      gpu::toLinearEncoding(tensorTy.getEncoding(), tensorTy.getShape());
+  auto linAttr = gpu::toLinearEncoding(tensorTy);
   auto order = linAttr.getOrder();
   unsigned align = getAlignment(value);
 
@@ -1336,8 +1335,7 @@ unsigned ModuleAxisInfoAnalysis::getAlignment(Value value) {
   auto *axisInfo = getAxisInfo(value);
   if (!axisInfo)
     return 1;
-  auto linAttr =
-      gpu::toLinearEncoding(tensorTy.getEncoding(), tensorTy.getShape());
+  auto linAttr = gpu::toLinearEncoding(tensorTy);
   auto order = linAttr.getOrder();
 
   // FIXME: should this be an assertion instead?
@@ -1378,8 +1376,7 @@ unsigned ModuleAxisInfoAnalysis::getMaskAlignment(Value mask) {
   auto *axisInfo = getAxisInfo(mask);
   if (!axisInfo)
     return 1;
-  auto linAttr =
-      gpu::toLinearEncoding(tensorTy.getEncoding(), tensorTy.getShape());
+  auto linAttr = gpu::toLinearEncoding(tensorTy);
 
   // FIXME: should this be an assertion instead?
   // Temporarily added to avoid crashing on some tests.