[all-commits] [llvm/llvm-project] 847710: [mlir][linalg][bufferize] Add dialect filter to Bu...

[Matthias Springer via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 847710f7b77ea4e3cd43f62b5b7d920ac47405a5
      https://github.com/llvm/llvm-project/commit/847710f7b77ea4e3cd43f62b5b7d920ac47405a5
  Author: Matthias Springer <springerm at google.com>
  Date:   2021-12-08 (Wed, 08 Dec 2021)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/ComprehensiveBufferize/BufferizableOpInterface.h
    M mlir/include/mlir/Dialect/Linalg/ComprehensiveBufferize/BufferizableOpInterface.td
    M mlir/lib/Dialect/Linalg/ComprehensiveBufferize/BufferizableOpInterface.cpp
    M mlir/lib/Dialect/Linalg/ComprehensiveBufferize/ComprehensiveBufferize.cpp
    M mlir/lib/Dialect/Linalg/ComprehensiveBufferize/LinalgInterfaceImpl.cpp
    M mlir/lib/Dialect/Linalg/ComprehensiveBufferize/TensorInterfaceImpl.cpp
    M mlir/test/Dialect/Linalg/comprehensive-module-bufferize-partial.mlir
    M mlir/test/lib/Dialect/Linalg/TestComprehensiveBufferize.cpp

  Log Message:
  -----------
  [mlir][linalg][bufferize] Add dialect filter to BufferizationOptions

This adds a new option `dialectFilter` to BufferizationOptions. Only ops from dialects that are allow-listed in the filter are bufferized. Other ops are left unbufferized. Note: This option requires `allowUnknownOps = true`.

To make use of `dialectFilter`, BufferizationOptions or BufferizationState must be passed to various helper functions.

The purpose of this change is to provide a better infrastructure for partial bufferization, which will be fully activated in a subsequent change.

Differential Revision: https://reviews.llvm.org/D114691