fix

Signed-off-by: Anatoly Myachev <[email protected]>

Author: anmyachev

lib/Dialect/TritonGPU/IR/Dialect.cpp

@@ -299,8 +299,12 @@ SmallVector<unsigned> getOrder(Attribute layout) {
   }
   if (auto dotLayout = dyn_cast<DotOperandEncodingAttr>(layout)) {
     auto rank = getWarpsPerCTA(dotLayout.getParent()).size();
-    auto mmaParent = dyn_cast<MmaEncodingTrait>(dotLayout.getParent());
-    return mmaParent.getOrderForDotOperand(dotLayout.getOpIdx(), rank);
+    if (auto mmaParent = dyn_cast<MmaEncodingTrait>(dotLayout.getParent())) {
+      return mmaParent.getOrderForDotOperand(dotLayout.getOpIdx(), rank);
+    }
+    // This branch had to be left because not all types
+    // inherit `MmaEncodingTrait` interface, for example `BlockedEncodingAttr`.
+    return getOrderForDotOperand(dotLayout.getOpIdx(), rank);
   }
   if (auto sliceLayout = dyn_cast<SliceEncodingAttr>(layout)) {
     SmallVector<unsigned> parentOrder = getOrder(sliceLayout.getParent());