[all-commits] [llvm/llvm-project] f2e6ca: [MLIR][Vector] Add warp distribution for `vector.s...

[Artem Kroviakov via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: f2e6ca805dbb8e55de480ddd2bc932c7e0fccf89
      https://github.com/llvm/llvm-project/commit/f2e6ca805dbb8e55de480ddd2bc932c7e0fccf89
  Author: Artem Kroviakov <71938912+akroviakov at users.noreply.github.com>
  Date:   2025-08-28 (Thu, 28 Aug 2025)

  Changed paths:
    M mlir/lib/Dialect/Vector/Transforms/VectorDistribute.cpp
    M mlir/test/Dialect/Vector/vector-warp-distribute.mlir

  Log Message:
  -----------
  [MLIR][Vector] Add warp distribution for `vector.step` op (#155425)

This PR adds a distribution pattern for
[`vector.step`](https://mlir.llvm.org/docs/Dialects/Vector/#vectorstep-vectorstepop)
op.

The result of the step op is a vector containing a sequence
`[0,1,...,N-1]`. For the warp distribution, we consider a vector with `N
== warp_size` (think SIMD). Distributing it to SIMT, means that each
lane is represented by a thread/lane id scalar.

More complex cases with the support for warp size multiples (e.g.,
`[0,1,...,2*N-1]`) require additional layout information to be handled
properly. Such support may be added later.

The lane id scalar is wrapped into a `vector<1xindex>` to emulate the
sequence distribution result.
Other than that, the distribution is similar to that of
`arith.constant`.



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