[mlir][bufferization] Fix OpFilter::denyDialect (#108249)

mfrancio · web-flow · commit aabb0121eece · 2024-09-11T12:03:49.000-07:00
The implementation would crash with unloaded dialects.
diff --git a/mlir/include/mlir/Dialect/Bufferization/IR/BufferizableOpInterface.h b/mlir/include/mlir/Dialect/Bufferization/IR/BufferizableOpInterface.h
@@ -152,7 +152,7 @@ class OpFilter {
   /// This function adds a DENY entry.
   void denyDialect(StringRef dialectNamespace) {
     Entry::FilterFn filterFn = [=](Operation *op) {
-      return op->getDialect()->getNamespace() == dialectNamespace;
+      return op->getName().getDialectNamespace() == dialectNamespace;
     };
     entries.push_back(Entry{filterFn, Entry::FilterType::DENY});
   }

Original file line number	Diff line number	Diff line change
`@@ -152,7 +152,7 @@ class OpFilter {`
`152`	`152`	`/// This function adds a DENY entry.`
`153`	`153`	`void denyDialect(StringRef dialectNamespace) {`
`154`	`154`	`Entry::FilterFn filterFn = [=](Operation *op) {`
`155`		`- return op->getDialect()->getNamespace() == dialectNamespace;`
	`155`	`+ return op->getName().getDialectNamespace() == dialectNamespace;`
`156`	`156`	`};`
`157`	`157`	`entries.push_back(Entry{filterFn, Entry::FilterType::DENY});`
`158`	`158`	`}`