Searched defs:ThreadTokenCallback (Results 1 – 1 of 1) sorted by relevance
46 struct GpuAsyncRegionPass::ThreadTokenCallback { struct in GpuAsyncRegionPass47 ThreadTokenCallback(MLIRContext &context) : builder(&context) {} in ThreadTokenCallback() function49 WalkResult operator()(Block *block) { in operator ()()66 LogicalResult visit(Operation *op) { in visit()87 LogicalResult rewriteAsyncOp(gpu::AsyncOpInterface asyncOp) { in rewriteAsyncOp()126 Value createWaitOp(Location loc, Type resultType, ValueRange operands) { in createWaitOp()130 OpBuilder builder;136 Value currentToken = {};