[mlir][bufferization] Fix OpFilter not to crash on unloaded dialects

d0k · d0k · commit 577306133e14 · 2024-03-19T11:40:58.000+01:00
diff --git a/mlir/include/mlir/Dialect/Bufferization/IR/BufferizableOpInterface.h b/mlir/include/mlir/Dialect/Bufferization/IR/BufferizableOpInterface.h
@@ -142,7 +142,7 @@ class OpFilter {
   /// This function adds an ALLOW entry.
   void allowDialect(StringRef dialectNamespace) {
     Entry::FilterFn filterFn = [=](Operation *op) {
-      return op->getDialect()->getNamespace() == dialectNamespace;
+      return op->getName().getDialectNamespace() == dialectNamespace;
     };
     entries.push_back(Entry{filterFn, Entry::FilterType::ALLOW});
   }

Original file line number	Diff line number	Diff line change
`@@ -142,7 +142,7 @@ class OpFilter {`
`142`	`142`	`/// This function adds an ALLOW entry.`
`143`	`143`	`void allowDialect(StringRef dialectNamespace) {`
`144`	`144`	`Entry::FilterFn filterFn = [=](Operation *op) {`
`145`		`- return op->getDialect()->getNamespace() == dialectNamespace;`
	`145`	`+ return op->getName().getDialectNamespace() == dialectNamespace;`
`146`	`146`	`};`
`147`	`147`	`entries.push_back(Entry{filterFn, Entry::FilterType::ALLOW});`
`148`	`148`	`}`