[all-commits] [llvm/llvm-project] 0b0520: [MLIR][LinAlg] Detensoring CF cost-model: look for...

[Kareem Ergawy via All-commits]

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

  Changed paths:
    M mlir/lib/Dialect/Linalg/Transforms/Detensorize.cpp
    A mlir/test/Dialect/Linalg/detensorize_0d.mlir
    M mlir/test/Dialect/Linalg/detensorize_if.mlir
    M mlir/test/Dialect/Linalg/detensorize_while.mlir
    R mlir/test/Dialect/Linalg/detensorized_0d.mlir

  Log Message:
  -----------
  [MLIR][LinAlg] Detensoring CF cost-model: look forward.

This patch extends the control-flow cost-model for detensoring by
implementing a forward-looking pass on block arguments that should be
detensored. This makes sure that if a (to-be-detensored) block argument
"escapes" its block through the terminator, then the successor arguments
are also detensored.

Reviewed By: silvas

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