[all-commits] [llvm/llvm-project] 0b17d4: [mlir][Linalg] Tile sizes for Conv ops vectorizati...

[Jakub Lichman via All-commits]

  Branch: refs/heads/master
  Home:   https://github.com/llvm/llvm-project
  Commit: 0b17d4754a94b7129c2483762acd586783802b12
      https://github.com/llvm/llvm-project/commit/0b17d4754a94b7129c2483762acd586783802b12
  Author: Jakub Lichman <limo at google.com>
  Date:   2020-09-30 (Wed, 30 Sep 2020)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h
    M mlir/integration_test/Dialect/Linalg/CPU/test-conv-1d-call.mlir
    M mlir/integration_test/Dialect/Linalg/CPU/test-conv-1d-ncw-call.mlir
    M mlir/integration_test/Dialect/Linalg/CPU/test-conv-1d-nwc-call.mlir
    M mlir/integration_test/Dialect/Linalg/CPU/test-conv-2d-call.mlir
    M mlir/integration_test/Dialect/Linalg/CPU/test-conv-2d-nchw-call.mlir
    M mlir/integration_test/Dialect/Linalg/CPU/test-conv-2d-nhwc-call.mlir
    M mlir/integration_test/Dialect/Linalg/CPU/test-conv-3d-call.mlir
    M mlir/integration_test/Dialect/Linalg/CPU/test-conv-3d-ncdhw-call.mlir
    M mlir/integration_test/Dialect/Linalg/CPU/test-conv-3d-ndhwc-call.mlir
    M mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp
    M mlir/test/Conversion/LinalgToVector/linalg-to-vector.mlir
    M mlir/test/lib/Transforms/TestConvVectorization.cpp

  Log Message:
  -----------
  [mlir][Linalg] Tile sizes for Conv ops vectorization added as pass arguments

Current setup for conv op vectorization does not enable user to specify tile
sizes as well as dimensions for vectorization. In this commit we change that by
adding tile sizes as pass arguments. Every dimension with corresponding tile
size > 1 is automatically vectorized.

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