maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

[mlir][bufferize] Add helpers for templatized DENY filters 

We already have templatized ALLOW filters but the DENY filters were missing.

Differential Revision: https://reviews.llvm.org/D125358
This commit is contained in:
Matthias Springer 2022-05-12 09:17:04 +02:00
parent 698fda0e3e
commit 011f1b1c1f
1 changed files with 28 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
mlir/include/mlir/Dialect/Bufferization/IR/BufferizableOpInterface.h
View File

@ -93,6 +93,15 @@ struct BufferizationOptions {
0, (allowDialectInFilterImpl<DialectTs>(), 0)...}; 0, (allowDialectInFilterImpl<DialectTs>(), 0)...};
} }
/// Deny the given dialects in the filter.
///
/// This function adds one or multiple DENY filters.
template <typename... DialectTs> void denyDialectInFilter() {
// FIXME: In c++17 this can be simplified by using 'fold expressions'.
(void)std::initializer_list<int>{
0, (denyDialectInFilterImpl<DialectTs>(), 0)...};
}
/// Allow the given dialect in the filter. /// Allow the given dialect in the filter.
/// ///
/// This function adds an ALLOW filter. /// This function adds an ALLOW filter.
@ -114,6 +123,15 @@ struct BufferizationOptions {
0, (allowOperationInFilterImpl<OpTys>(), 0)...}; 0, (allowOperationInFilterImpl<OpTys>(), 0)...};
} }
/// Deny the given ops in the filter.
///
/// This function adds one or multiple DENY filters.
template <typename... OpTys> void denyOperationInFilter() {
// FIXME: In c++17 this can be simplified by using 'fold expressions'.
(void)std::initializer_list<int>{
0, (denyOperationInFilterImpl<OpTys>(), 0)...};
}
/// Allow the given op in the filter. /// Allow the given op in the filter.
/// ///
/// This function adds an ALLOW filter. /// This function adds an ALLOW filter.
@ -249,11 +267,21 @@ private:
allowDialectInFilter(DialectT::getDialectNamespace()); allowDialectInFilter(DialectT::getDialectNamespace());
} }
/// Deny a dialect.
template <typename DialectT> void denyDialectInFilterImpl() {
denyDialectInFilter(DialectT::getDialectNamespace());
}
/// Allow an op. /// Allow an op.
template <typename OpTy> template <typename OpTy>
void allowOperationInFilterImpl() { void allowOperationInFilterImpl() {
allowOperationInFilter(OpTy::getOperationName()); allowOperationInFilter(OpTy::getOperationName());
} }
/// Deny an op.
template <typename OpTy> void denyOperationInFilterImpl() {
denyOperationInFilter(OpTy::getOperationName());
}
}; };
/// Specify fine-grain relationship between buffers to enable more analysis. /// Specify fine-grain relationship between buffers to enable more analysis.