[Mlir-commits] [mlir] d538c56 - [mlir][bufferize] Fix broken implementation of denyDialect (#85881)

[llvmlistbot at llvm.org]

Author: Matteo Franciolini
Date: 2024-03-19T23:28:20-07:00
New Revision: d538c5616f4d25c6f50a1285984aba806252ce17

URL: https://github.com/llvm/llvm-project/commit/d538c5616f4d25c6f50a1285984aba806252ce17
DIFF: https://github.com/llvm/llvm-project/commit/d538c5616f4d25c6f50a1285984aba806252ce17.diff

LOG: [mlir][bufferize] Fix broken implementation of denyDialect (#85881)

The current implementation is not functional and does not compile. The
patch allows to correctly mark a dialect as denied for the purpose of
bufferization.

Added: 
    

Modified: 
    mlir/include/mlir/Dialect/Bufferization/IR/BufferizableOpInterface.h

Removed: 
    


################################################################################
diff  --git a/mlir/include/mlir/Dialect/Bufferization/IR/BufferizableOpInterface.h b/mlir/include/mlir/Dialect/Bufferization/IR/BufferizableOpInterface.h
index 94d5c08dfe4575..3a61a4b34765e2 100644
--- a/mlir/include/mlir/Dialect/Bufferization/IR/BufferizableOpInterface.h
+++ b/mlir/include/mlir/Dialect/Bufferization/IR/BufferizableOpInterface.h
@@ -147,6 +147,16 @@ class OpFilter {
     entries.push_back(Entry{filterFn, Entry::FilterType::ALLOW});
   }
 
+  /// Deny the given dialect.
+  ///
+  /// This function adds a DENY entry.
+  void denyDialect(StringRef dialectNamespace) {
+    Entry::FilterFn filterFn = [=](Operation *op) {
+      return op->getDialect()->getNamespace() == dialectNamespace;
+    };
+    entries.push_back(Entry{filterFn, Entry::FilterType::DENY});
+  }
+
   /// Allow the given ops.
   ///
   /// This function adds one or multiple ALLOW entries.