102   bool getKeyAMXShapes(IntrinsicInst *KeyAMX, SmallVector<Value *, 8> &Shapes);
109                           IntrinsicInst *KeyAMX);
217                                          IntrinsicInst *KeyAMX) {  in checkVolatileModel()  argument
221   if (!KeyAMX)  in checkVolatileModel()
226   for (Value *Op : KeyAMX->operands()) {  in checkVolatileModel()
234   return Loads.empty() && (ST == cast<Value>(KeyAMX));  in checkVolatileModel()
237 bool X86PreAMXConfig::getKeyAMXShapes(IntrinsicInst *KeyAMX,  in getKeyAMXShapes()  argument
239   for (unsigned I = 0; I < KeyAMX->getNumOperands(); I++) {  in getKeyAMXShapes()
240     Value *Op = KeyAMX->getOperand(I);  in getKeyAMXShapes()
249   if (!isTileStore(KeyAMX)) {  in getKeyAMXShapes()
250     Shapes.push_back(KeyAMX->getOperand(0));  in getKeyAMXShapes()
251     Shapes.push_back(KeyAMX->getOperand(1));  in getKeyAMXShapes()
270   IntrinsicInst *KeyAMX = nullptr;  in getShapesAndConfigPosEnd()  local
285       if (!checkVolatileModel(Loads, II, KeyAMX))  in getShapesAndConfigPosEnd()
290       assert(!KeyAMX && "Too many key amx intrinsic!");  in getShapesAndConfigPosEnd()
291       KeyAMX = II;  in getShapesAndConfigPosEnd()
297   if (!KeyAMX)  in getShapesAndConfigPosEnd()
298     KeyAMX = dyn_cast<IntrinsicInst>(&*PosEnd);  in getShapesAndConfigPosEnd()
302   getKeyAMXShapes(KeyAMX, Shapes);  in getShapesAndConfigPosEnd()