Lines Matching defs:frag

236     auto &frag = frags[fragIdx];
241       frag.regs.push_back(this->getOperand(operandIdx));
249       ignoreAttrNames.push_back(frag.ptxTypeAttr);
252   auto printMmaOperand = [&](const OperandFragment &frag) -> void {
253     p << " " << frag.operandName;
255     p.printOperands(frag.regs);
259   for (const auto &frag : frags) {
260     printMmaOperand(frag);
345                              OperandFragment &frag) -> LogicalResult {
349             .parseOperandList(frag.regs, OpAsmParser::Delimiter::OptionalSquare)
381     auto &frag = frags[iter.index()];
382     frag.regTypes.resize(frag.regs.size(), iter.value());
383     if (failed(parser.resolveOperands(frag.regs, frag.regTypes,
386     frag.elemtype =
387         inferOperandMMAType(frag.regTypes[0], /*isAccum=*/iter.index() < 2);
398     const auto &frag = frags[idx];
400     if (!frag.elemtype.has_value() && !attr.has_value()) {
408           names[idx], MMATypesAttr::get(parser.getContext(), *frag.elemtype));
627                                                    NVVM::MMAFrag frag, int nRow,
636     if (frag == NVVM::MMAFrag::a || frag == NVVM::MMAFrag::b)
649     if (frag == NVVM::MMAFrag::a)
651     if (frag == NVVM::MMAFrag::b)
671 inferMMATypeFromMNK(NVVM::MMATypes type, NVVM::MMAFrag frag, int m, int n,
674   if (frag == NVVM::MMAFrag::a) {
677   } else if (frag == NVVM::MMAFrag::b) {
685   return inferMMAType(type, frag, nRow, nCol, context);