From 63e6a859014f055496b007dc462f3f1f7eb91701 Mon Sep 17 00:00:00 2001
From: Yan Da <dyanab@connect.ust.hk>
Date: Fri, 15 Jul 2022 15:19:11 +0800
Subject: [PATCH] Fix blocked layout parser

---
 lib/Dialect/TritonGPU/IR/Dialect.cpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/lib/Dialect/TritonGPU/IR/Dialect.cpp b/lib/Dialect/TritonGPU/IR/Dialect.cpp
index 9a99e8ae9..95c8c7b30 100644
--- a/lib/Dialect/TritonGPU/IR/Dialect.cpp
+++ b/lib/Dialect/TritonGPU/IR/Dialect.cpp
@@ -65,6 +65,9 @@ static Attribute parseBlocked(AsmParser &parser, Type type) {
     } else if (attr.getName() == "order") {
       if (parseIntArrayAttr(parser, attr, order, "order").failed())
         return {};
+    } else if (attr.getName() == "broadcastAxis") {
+      if (parseIntArrayAttr(parser, attr, broadcastAxis, "broadcastAxis").failed())
+        return {};
     } else {
       parser.emitError(parser.getNameLoc(), "unexpected key: ")
           << attr.getName().strref();