xiaoleis-nv fbf81e3004

Enable attaching LLVM loop annotations to scf.for (#102562) ...

We recently discovered that the loop with a dynamic upper bound is
unexpectedly unrolled during the NVVM to PTX process. By attaching the
`llvm.loop_annotation`, we can control the unrolling behavior precisely.

This PR enables the `cf.cond_br` to retain the loop annotation of
`scf.for` after the `convert-scf-to-cf` pass. This change allows users
to have precise control over the loop behavior during backend
transformation.

---------

Co-authored-by: Xiaolei Shi <xiaoleis@nvidia.com>

2024-08-12 18:25:13 +02:00

..

convert-to-cfg.mlir

Enable attaching LLVM loop annotations to scf.for (#102562)

2024-08-12 18:25:13 +02:00