Skip to content

Plumbing the topk to the nvFuser executor#2237

Merged
t-vi merged 4 commits intomainfrom
pbasu_topk
Jun 17, 2025
Merged

Plumbing the topk to the nvFuser executor#2237
t-vi merged 4 commits intomainfrom
pbasu_topk

Conversation

@protonu
Copy link
Copy Markdown
Collaborator

@protonu protonu commented Jun 13, 2025

What does this PR do?

This builds on the PR NVIDIA/Fuser#4605 to plumb execution of top-K to the nvFuser executor backend. Existing topK tests in test_ops.py should test out this path.

@protonu protonu marked this pull request as draft June 13, 2025 19:50
@protonu protonu requested a review from jjsjann123 June 13, 2025 19:50
@protonu protonu marked this pull request as ready for review June 16, 2025 18:12
@protonu
Copy link
Copy Markdown
Collaborator Author

protonu commented Jun 16, 2025

This now works as NVIDIA/Fuser#4629 has landed.

Copy link
Copy Markdown
Collaborator

@t-vi t-vi left a comment

Choose a reason for hiding this comment

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

Thank you @protonu

@t-vi t-vi merged commit b3e062a into main Jun 17, 2025
52 checks passed
@t-vi t-vi deleted the pbasu_topk branch June 17, 2025 09:01
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.

2 participants