This is an archive of the discontinued LLVM Phabricator instance.

Show First 20 Lines • Show All 352 Lines • ▼ Show 20 Lines	BufferizationRewriter(MLIRContext ctx, DenseSet<Operation > &erasedOps,
: IRRewriter(ctx), erasedOps(erasedOps), toMemrefOps(toMemrefOps),		: IRRewriter(ctx), erasedOps(erasedOps), toMemrefOps(toMemrefOps),
worklist(worklist), analysisState(options), opFilter(opFilter),		worklist(worklist), analysisState(options), opFilter(opFilter),
statistics(statistics) {		statistics(statistics) {
setListener(this);		setListener(this);
}		}

protected:		protected:
void notifyOperationRemoved(Operation *op) override {		void notifyOperationRemoved(Operation *op) override {
		// TODO: Walk can be removed when D144193 has landed.
		op->walk([&](Operation *op) {
erasedOps.insert(op);		erasedOps.insert(op);
// Erase if present.		// Erase if present.
toMemrefOps.erase(op);		toMemrefOps.erase(op);
		});
}		}

void notifyOperationInserted(Operation *op) override {		void notifyOperationInserted(Operation *op) override {
erasedOps.erase(op);		erasedOps.erase(op);

// Gather statistics about allocs and deallocs.		// Gather statistics about allocs and deallocs.
if (statistics) {		if (statistics) {
if (auto sideEffectingOp = dyn_cast<MemoryEffectOpInterface>(op)) {		if (auto sideEffectingOp = dyn_cast<MemoryEffectOpInterface>(op)) {
▲ Show 20 Lines • Show All 182 Lines • Show Last 20 Lines

[mlir][bufferize] Fix tracking of erased opsClosedPublic