arun-thmn wrote: > Could you also add a few test cases with loops to exercise the new logic around iter args etc.? Added of couple of unit tests with loops for both AVX512 and AVX2. https://github.com/llvm/llvm-project/pull/174590