Skip to content

Commit

Permalink
nit rename step
Browse files Browse the repository at this point in the history
Signed-off-by: Stanley Winata <[email protected]>
  • Loading branch information
raikonenfnu committed Oct 24, 2024
1 parent 8ce9b09 commit 5d4d0e3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion iree/turbine/kernel/wave/decompose_reduce_ops.py
Original file line number Diff line number Diff line change
Expand Up @@ -80,7 +80,7 @@ def decompose_reduce_ops(
1. Source Reduce: Each thread reduce locally all it's sources.
2. Local Reduce: Each thread reduces all elements carried by it along
the reduction dimensions.
3. Thread Reduce: Each thread reduces result of step 1 across threads
3. Thread Reduce: Each thread reduces result of step 2 across threads
by doing a butterfly shuffle.
4. Accumulator Reduce: Each thread reduces it's intermediate reduced
results with the accumulator it holds.
Expand Down

0 comments on commit 5d4d0e3

Please sign in to comment.