[all-commits] [llvm/llvm-project] 9931ee: [mlir][vector] Fix FlattenGather for scalable vect...

[Cullen Rhodes via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 9931ee61d99c101db653ae21706f1edce4b39781
      https://github.com/llvm/llvm-project/commit/9931ee61d99c101db653ae21706f1edce4b39781
  Author: Cullen Rhodes <cullen.rhodes at arm.com>
  Date:   2024-06-24 (Mon, 24 Jun 2024)

  Changed paths:
    M mlir/lib/Dialect/Vector/Transforms/LowerVectorGather.cpp
    M mlir/test/Dialect/Vector/vector-gather-lowering.mlir

  Log Message:
  -----------
  [mlir][vector] Fix FlattenGather for scalable vectors (#96074)

This pattern flattens vector.gather ops by unrolling the outermost
dimension for rank > 2 vectors. There's two issues with this pattern for
scalable vectors:

  1. The unrolling doesn't take vscale into account. A constraint is
     added to disable this pattern for vectors with leading scalable
     dims.
  2. The scalable dims are dropped when creating the new gather. Fixed
     by propagating the flags.

Depends on #96049.



To unsubscribe from these emails, change your notification settings at https://github.com/llvm/llvm-project/settings/notifications