Fix AxisAnalysis to avoid out of bound access. (#3832)

Resolves issue #3830.

---------

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

Author: etiotto

third_party/intel/lib/Analysis/AxisInfo.cpp

@@ -1236,6 +1236,13 @@ unsigned ModuleAxisInfoAnalysis::getAlignment(Value value) {
   auto linAttr =
       gpu::toLinearEncoding(tensorTy.getEncoding(), tensorTy.getShape());
   auto order = linAttr.getOrder();
+
+  // FIXME: should this be an assertion instead?
+  // Temporarily added to avoid crashing on some tests.
+  // See issue #3842.
+  if (order[0] >= axisInfo->getRank())
+    return 1;
+
   auto maxMultipleBytes = axisInfo->getDivisibility(order[0]);
   auto maxContig = axisInfo->getContiguity(order[0]);
 
@@ -1270,7 +1277,14 @@ unsigned ModuleAxisInfoAnalysis::getMaskAlignment(Value mask) {
     return 1;
   auto linAttr =
       gpu::toLinearEncoding(tensorTy.getEncoding(), tensorTy.getShape());
+
+  // FIXME: should this be an assertion instead?
+  // Temporarily added to avoid crashing on some tests.
+  // See issue #3842.
   auto maskOrder = linAttr.getOrder();
+  if (maskOrder[0] >= axisInfo->getRank())
+    return 1;
+
   auto alignment = std::max<unsigned>(axisInfo->getConstancy(maskOrder[0]), 1);
   LDBG("getMaskAlignment maskOrder[0] " << maskOrder[0] << " alignment "
                                         << alignment);