.

2022-12-08 21:36:47 -08:00
parent 13644e7ac4
commit fa6dbbff60
2 changed files with 4 additions and 2 deletions
--- a/lib/Dialect/TritonGPU/IR/Dialect.cpp
+++ b/lib/Dialect/TritonGPU/IR/Dialect.cpp
@@ -593,6 +593,8 @@ Attribute DotOperandEncodingAttr::parse(AsmParser &parser, Type type) {
  if(parent.isa<MmaEncodingAttr>() &&
     parent.cast<MmaEncodingAttr>().getVersion() == 1){
    isMMAv1Row = attrs.get("isMMAv1Row");
+    if(!isMMAv1Row)
+      llvm::report_fatal_error("isMMAv1Row attribute is missing");
  }
  return parser.getChecked<DotOperandEncodingAttr>(parser.getContext(), opIdx,
                                                   parent, isMMAv1Row);