[all-commits] [llvm/llvm-project] aa6eb2: [MLIR][LinAlg] Implement detensoring cost-modelling.

[Kareem Ergawy via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: aa6eb2af10094e427827343b67b25d606dde10b7
      https://github.com/llvm/llvm-project/commit/aa6eb2af10094e427827343b67b25d606dde10b7
  Author: KareemErgawy-TomTom <kareem.ergawy at gmail.com>
  Date:   2021-04-13 (Tue, 13 Apr 2021)

  Changed paths:
    M mlir/include/mlir/Dialect/StandardOps/Transforms/FuncConversions.h
    M mlir/include/mlir/Transforms/DialectConversion.h
    M mlir/lib/Dialect/Linalg/Transforms/Detensorize.cpp
    M mlir/lib/Dialect/StandardOps/Transforms/FuncConversions.cpp
    M mlir/lib/Transforms/Utils/DialectConversion.cpp
    A mlir/test/Dialect/Linalg/detensorize_if.mlir
    A mlir/test/Dialect/Linalg/detensorize_trivial.mlir
    A mlir/test/Dialect/Linalg/detensorize_while.mlir
    A mlir/test/Dialect/Linalg/detensorize_while_failure.mlir
    A mlir/test/Dialect/Linalg/detensorize_while_pure_cf.mlir
    M mlir/test/Dialect/Linalg/detensorized_0d.mlir
    R mlir/test/Dialect/Linalg/detensorized_while.mlir

  Log Message:
  -----------
  [MLIR][LinAlg] Implement detensoring cost-modelling.

This patch introduces the neccessary infrastructure changes to implement
cost-modelling for detensoring. In particular, it introduces the
following changes:
- An extension to the dialect conversion framework to selectively
convert sub-set of non-entry BB arguments.
- An extension to branch conversion pattern to selectively convert
sub-set of a branche's operands.
- An interface for detensoring cost-modelling.
- 2 simple implementations of 2 different cost models.

This sets the stage to explose cost-modelling for detessoring in an
easier way. We still need to come up with better cost models.

Reviewed By: silvas

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