[all-commits] [llvm/llvm-project] c90877: [mlir] Fix masked vector transfer ops with broadcasts

Matthias Springer via All-commits all-commits at lists.llvm.org
Wed May 12 19:38:45 PDT 2021


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: c9087788f7e41285445729127dd07ff7f82e3fc0
      https://github.com/llvm/llvm-project/commit/c9087788f7e41285445729127dd07ff7f82e3fc0
  Author: Matthias Springer <springerm at google.com>
  Date:   2021-05-13 (Thu, 13 May 2021)

  Changed paths:
    M mlir/include/mlir/Interfaces/VectorInterfaces.h
    M mlir/include/mlir/Interfaces/VectorInterfaces.td
    M mlir/lib/Conversion/VectorToSCF/ProgressiveVectorToSCF.cpp
    M mlir/lib/Dialect/Vector/VectorOps.cpp
    M mlir/lib/Interfaces/VectorInterfaces.cpp
    M mlir/test/Integration/Dialect/Vector/CPU/test-transfer-read-1d.mlir
    M mlir/test/Integration/Dialect/Vector/CPU/test-transfer-read-2d.mlir
    M mlir/test/Integration/Dialect/Vector/CPU/test-transfer-read-3d.mlir

  Log Message:
  -----------
  [mlir] Fix masked vector transfer ops with broadcasts

Broadcast dimensions of a vector transfer op have no corresponding dimension in the mask vector. E.g., a 2-D TransferReadOp, where one dimension is a broadcast, can have a 1-D `mask` attribute.

This commit also adds a few additional transfer op integration tests for various combinations of broadcasts, masking, dim transposes, etc.

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




More information about the All-commits mailing list