Searched defs:ThreadTokenCallback (Results 1 – 1 of 1) sorted by relevance
44 struct GpuAsyncRegionPass::ThreadTokenCallback { struct in GpuAsyncRegionPass45 ThreadTokenCallback(MLIRContext &context) : builder(&context) {} in ThreadTokenCallback() function55 WalkResult operator()(Operation *op) { in operator ()()73 LogicalResult rewriteAsyncOp(gpu::AsyncOpInterface asyncOp) { in rewriteAsyncOp()105 Value createWaitOp(Location loc, Type resultType, ValueRange operands) { in createWaitOp()109 OpBuilder builder;110 const Type tokenType = builder.getType<gpu::AsyncTokenType>();115 Value currentToken = {};