Lines Matching refs:KeyAMX
103 bool getKeyAMXShapes(IntrinsicInst *KeyAMX, SmallVector<Value *, 8> &Shapes);
110 IntrinsicInst *KeyAMX);
208 IntrinsicInst *KeyAMX) { in checkVolatileModel() argument
212 if (!KeyAMX) in checkVolatileModel()
217 for (Value *Op : KeyAMX->operands()) { in checkVolatileModel()
225 return Loads.empty() && (ST == cast<Value>(KeyAMX)); in checkVolatileModel()
228 bool X86PreAMXConfig::getKeyAMXShapes(IntrinsicInst *KeyAMX, in getKeyAMXShapes() argument
230 for (unsigned I = 0; I < KeyAMX->getNumOperands(); I++) { in getKeyAMXShapes()
231 Value *Op = KeyAMX->getOperand(I); in getKeyAMXShapes()
240 if (!isTileStore(KeyAMX)) { in getKeyAMXShapes()
241 Shapes.push_back(KeyAMX->getOperand(0)); in getKeyAMXShapes()
242 Shapes.push_back(KeyAMX->getOperand(1)); in getKeyAMXShapes()
261 IntrinsicInst *KeyAMX = nullptr; in getShapesAndConfigPosEnd() local
276 if (!checkVolatileModel(Loads, II, KeyAMX)) in getShapesAndConfigPosEnd()
281 assert(!KeyAMX && "Too many key amx intrinsic!"); in getShapesAndConfigPosEnd()
282 KeyAMX = II; in getShapesAndConfigPosEnd()
288 if (!KeyAMX) in getShapesAndConfigPosEnd()
289 KeyAMX = dyn_cast<IntrinsicInst>(&*PosEnd); in getShapesAndConfigPosEnd()
293 getKeyAMXShapes(KeyAMX, Shapes); in getShapesAndConfigPosEnd()