Searched refs:foreachThreadOp (Results 1 – 3 of 3) sorted by relevance
175 if (scf::ForeachThreadOp foreachThreadOp = in matchAndRewrite() local177 for (int64_t idx = 0; idx < foreachThreadOp.getRank(); ++idx) { in matchAndRewrite()178 if (foreachThreadOp.getThreadIndices()[idx] == iv) { in matchAndRewrite()180 ub = foreachThreadOp.getNumThreads()[idx]; in matchAndRewrite()
925 getInsertionDest(ForeachThreadOp foreachThreadOp) { in getInsertionDest() argument926 PerformConcurrentlyOp terminator = foreachThreadOp.getTerminator(); in getInsertionDest()945 auto foreachThreadOp = cast<ForeachThreadOp>(op); in getAliasingOpOperand() local946 return {getInsertionDest(foreachThreadOp)[opResult.getResultNumber()]}; in getAliasingOpOperand()964 auto foreachThreadOp = cast<ForeachThreadOp>(op); in bufferize() local968 for (OpResult opResult : foreachThreadOp->getOpResults()) in bufferize()977 foreachThreadOp.getLoc(), newResultTypes, in bufferize()978 foreachThreadOp.getNumThreads(), in bufferize()979 extractFromI64ArrayAttr(foreachThreadOp.getThreadDimMapping())); in bufferize()983 rewriter.mergeBlocks(foreachThreadOp.getBody(), in bufferize()
252 scf::ForeachThreadOp foreachThreadOp = b.create<scf::ForeachThreadOp>( in tileToForeachThreadOpImpl() local257 b.setInsertionPointToStart(foreachThreadOp.getBody(0)); in tileToForeachThreadOpImpl()258 ValueRange threadIds = foreachThreadOp.getThreadIndices(); in tileToForeachThreadOpImpl()327 b.setInsertionPointToStart(foreachThreadOp.getTerminator().getBody()); in tileToForeachThreadOpImpl()334 return ForeachThreadTilingResult{foreachThreadOp, tiledOp}; in tileToForeachThreadOpImpl()