commit 0f21989438e5c422287882cb67a315329b7b744e
author Krzysztof Drewniak <Krzysztof.Drewniak@amd.com> Mon Dec 02 18:15:37 2024 -0600
committer GitHub <noreply@github.com> Mon Dec 02 18:15:37 2024 -0600
tree d4b23e996df324c8f339688a6c88ecf27651ee4a
parent 4ee7d600c9ef024f85c7f6860abdebc1a964f7bc

[Codegen][GPU] Add range information to GPU dispatch IDs (#17707)

First, this patch implements InferIntRangeInterface for
hal.interface.workgroup.{size,id,count} using a local upper_bound
attribute.

Then, it adds a -iree-codegen-gpu-propagate-dispatch-size-bounds pass
that adds these upper_bounds identifiers to the interface.workgroup
operations and to gpu.thread_id based on static information available
late in the codegen pipeline.

Then, it uses -optimize-int-arithmetic to optimize indexing after
-lower-affine, getting rid of a bunch of "if the input's negative" logic
that isn't actually needed in many of our kernels.

It also ensures that these upper_bound values propagate to LLVM.

---------

Signed-off-by: Krzysztof Drewniak <Krzysztof.Drewniak@amd.com>
Co-authored-by: Jakub Kuderski <kubakuderski@gmail.com>