#13901: Wide Reductions with Non-8-Tile Multiples

[wransom-TT]

Ticket

Link to Github Issue

Problem description

MaxPool2D should support wide reductions with non-max tile multiple C dimensions. This PR implements this feature for non-large kernels.

What's changed

The number of C tiles has been adjusted to use ceiling instead of a normal divide. The reader kernel has been updated to read either the max bytes per reduction, or a small remainder. The compute kernel has been updated to process max tile multiples for the first N-1 blocks, followed by a smaller number of tiles for the last block.

Currently large kernels will not work with non-max tile multiples so test skips have also been added to bypass these combinations with the new non-multiple test cases.

Checklist

• Post commit CI passes
• [N/A] Blackhole Post commit (if applicable)
• [N/A] Model regression CI testing passes (if applicable)
• [N/A] Device performance regression CI testing passes (if applicable)
• [N/A] (For models and ops writers) Full new models tests passes
• New/Existing tests provide coverage for changes