Lines Matching defs:callOp

368   void convertCallOp(A callOp, mlir::FunctionType fnTy) {
369     auto loc = callOp.getLoc();
370     rewriter->setInsertionPoint(callOp);
380       if (!callOp.getCallee()) {
383         newOpers.push_back(callOp.getOperand(0));
415       operands = callOp.getKernelOperands();
417       operands = callOp.getOperands().drop_front(dropFront);
470                 if (callOp.getCallee()) {
478                       *callOp.getCallee());
506               if (callOp.getPassArgPos() && *callOp.getPassArgPos() == index)
507                 passArgShift = newOpers.size() - *callOp.getPassArgPos();
523           loc, callOp.getKernel(), callOp.getGridSizeOperandValues(),
524           callOp.getBlockSizeOperandValues(),
525           callOp.getDynamicSharedMemorySize(), newOpers);
526       if (callOp.getClusterSizeX())
527         newCall.getClusterSizeXMutable().assign(callOp.getClusterSizeX());
528       if (callOp.getClusterSizeY())
529         newCall.getClusterSizeYMutable().assign(callOp.getClusterSizeY());
530       if (callOp.getClusterSizeZ())
531         newCall.getClusterSizeZMutable().assign(callOp.getClusterSizeZ());
535       if (callOp.getCallee()) {
537             rewriter->create<A>(loc, *callOp.getCallee(), newResTys, newOpers);
546             callOp.getContext(),
557           loc, newResTys, rewriter->getStringAttr(callOp.getMethod()),
558           callOp.getOperands()[0], newOpers,
559           rewriter->getI32IntegerAttr(*callOp.getPassArgPos() + passArgShift),
560           callOp.getProcedureAttrsAttr());
577           rewriter->setInsertionPointAfter(callOp);
581       replaceOp(callOp, newCallResults);