llvm-project

shylie/llvm-project

Fork 0

Commit Graph

Author	SHA1	Message	Date
Jakub Kuderski	c0345b4648	[mlir][gpu] Add subgroup_reduce to shuffle lowering (#76530 ) This supports both the scalar and the vector multi-reduction cases.	2024-01-02 16:14:22 -05:00
Jakub Kuderski	2af186f9bd	[mlir][gpu] Add patterns to break down subgroup reduce (#76271 ) The new patterns break down subgroup reduce ops with vector values into a sequence of subgroup reductions that fit the native shuffle size. The maximum/native shuffle size is parametrized. The overall goal is to be able to perform multi-element reductions with a sequence of `gpu.shuffle` ops.	2023-12-28 14:39:46 -05:00

Author

SHA1

Message

Date

Jakub Kuderski

c0345b4648

[mlir][gpu] Add subgroup_reduce to shuffle lowering (#76530 )

This supports both the scalar and the vector multi-reduction cases.

2024-01-02 16:14:22 -05:00

Jakub Kuderski

2af186f9bd

[mlir][gpu] Add patterns to break down subgroup reduce (#76271 )

The new patterns break down subgroup reduce ops with vector values into
a sequence of subgroup reductions that fit the native shuffle size. The
maximum/native shuffle size is parametrized.

The overall goal is to be able to perform multi-element reductions with
a sequence of `gpu.shuffle` ops.

2023-12-28 14:39:46 -05:00

2 Commits