Home
last modified time | relevance | path

Searched refs:executeOp (Results 1 – 2 of 2) sorted by relevance

/external/llvm-project/mlir/lib/Dialect/GPU/Transforms/
DAsyncRegionRewriter.cpp124 void operator()(async::ExecuteOp executeOp) { in operator ()()
125 if (!areAllUsersExecuteOrAwait(executeOp.token())) in operator ()()
128 for (auto &op : llvm::reverse(executeOp.getBody()->without_terminator())) { in operator ()()
143 auto executeOp = waitOp->getParentOfType<async::ExecuteOp>(); in ~DeferWaitCallback() local
147 auto &yieldOp = executeOp.getBody()->getOperations().back(); in ~DeferWaitCallback()
151 auto asyncTokens = addAsyncTokenResults(executeOp, numDependencies); in ~DeferWaitCallback()
154 for (Operation *user : executeOp.token().getUsers()) in ~DeferWaitCallback()
161 static ValueRange addAsyncTokenResults(async::ExecuteOp &executeOp, in addAsyncTokenResults()
163 auto numResults = executeOp.getNumResults() + count; in addAsyncTokenResults()
168 copy(executeOp.getResultTypes(), std::back_inserter(resultTypes)); in addAsyncTokenResults()
[all …]
/external/llvm-project/mlir/lib/Dialect/Async/IR/
DAsync.cpp105 auto executeOp = op->getParentOfType<ExecuteOp>(); in verify() local
106 auto types = llvm::map_range(executeOp.results(), [](const OpResult &result) { in verify()