Skip to content

[Codegen] Migrate ArgCompareOp to VectorizableOpInterface#23656

Merged
hanhanW merged 1 commit intomainfrom
users/hanhanW/vec-iface-c2a
Mar 10, 2026
Merged

[Codegen] Migrate ArgCompareOp to VectorizableOpInterface#23656
hanhanW merged 1 commit intomainfrom
users/hanhanW/vec-iface-c2a

Conversation

@hanhanW
Copy link
Contributor

@hanhanW hanhanW commented Mar 5, 2026

  • Add external model for IREE::LinalgExt::ArgCompareOp, inlining the vectorization logic from vectorizeLinalgExtArgCompare.
  • Remove ArgCompareOp from GenericVectorization TypeSwitch and candidate isa<> check (now dispatched through interface).

No new tests because it is an NFC in terms of functionality. It just follows different mechanism for vectorization.

It is a step towards https://lists.lfaidata.foundation/g/iree-technical-discussion/message/15

Assisted-by: Claude

@hanhanW hanhanW requested a review from bangtianliu March 5, 2026 01:50
@hanhanW hanhanW force-pushed the users/hanhanW/vec-iface-c2a branch 2 times, most recently from d8935f8 to 5e497e7 Compare March 5, 2026 03:31
Copy link
Contributor

@bangtianliu bangtianliu left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, Thanks!

@hanhanW
Copy link
Contributor Author

hanhanW commented Mar 6, 2026

Base automatically changed from users/hanhanW/vec-iface-c1 to main March 10, 2026 16:39
- Add external model for IREE::LinalgExt::ArgCompareOp, inlining the
  vectorization logic from vectorizeLinalgExtArgCompare.
- Remove ArgCompareOp from GenericVectorization TypeSwitch and
  candidate isa<> check (now dispatched through interface).

Signed-off-by: hanhanW <[email protected]>
@hanhanW hanhanW force-pushed the users/hanhanW/vec-iface-c2a branch from c5f784e to a4dc64b Compare March 10, 2026 16:51
@hanhanW hanhanW enabled auto-merge (squash) March 10, 2026 16:51
@hanhanW hanhanW merged commit 88f4211 into main Mar 10, 2026
84 of 87 checks passed
@hanhanW hanhanW deleted the users/hanhanW/vec-iface-c2a branch March 10, 2026 18:30
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants