LLITCHEV wrote: > I think you also need to refactor `isValidMaskedInputVector` for your topk implementation? I thought to do two separate PRs, but there is no reason why not to do it in one. Thanks! https://github.com/llvm/llvm-project/pull/89119