gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix blocked layout parser

Browse Source
This commit is contained in:
Yan Da
2022-07-15 15:19:11 +08:00
parent 65237f6117
commit 63e6a85901
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lib/Dialect/TritonGPU/IR/Dialect.cpp
View File

@@ -65,6 +65,9 @@ static Attribute parseBlocked(AsmParser &parser, Type type) {
} else if (attr.getName() == "order") { } else if (attr.getName() == "order") {
if (parseIntArrayAttr(parser, attr, order, "order").failed()) if (parseIntArrayAttr(parser, attr, order, "order").failed())
return {}; return {};
} else if (attr.getName() == "broadcastAxis") {
if (parseIntArrayAttr(parser, attr, broadcastAxis, "broadcastAxis").failed())
return {};
} else { } else {
parser.emitError(parser.getNameLoc(), "unexpected key: ") parser.emitError(parser.getNameLoc(), "unexpected key: ")
<< attr.getName().strref(); << attr.getName().strref();