Searched defs:waitOp (Results 1 – 6 of 6) sorted by relevance
76 if (auto waitOp = llvm::dyn_cast<gpu::WaitOp>(op)) { in visit() local 197 if (auto waitOp = dyn_cast<gpu::WaitOp>(op)) { in operator ()() local 210 auto waitOp = worklist[i]; in ~DeferWaitCallback() local 285 auto waitOp = builder.create<gpu::WaitOp>(loc, Type{}, tokens); addAsyncDependencyAfter() local [all...]
151 static bool checkTagMatch(AffineDmaStartOp startOp, AffineDmaWaitOp waitOp) { in checkTagMatch() argument
861 matchAndRewrite(gpu::WaitOp waitOp,OpAdaptor adaptor,ConversionPatternRewriter & rewriter) const matchAndRewrite() argument 891 matchAndRewrite(gpu::WaitOp waitOp,OpAdaptor adaptor,ConversionPatternRewriter & rewriter) const matchAndRewrite() argument [all...]
254 auto waitOp = dyn_cast<nvgpu::DeviceAsyncWaitOp>(op); in setAsyncWaitGroupsInFlight() local
2029 auto waitOp = value.getDefiningOp<WaitOp>(); matchAndRewrite() local
3379 mlir::acc::WaitOp waitOp = createSimpleOp<mlir::acc::WaitOp>( genACC() local