Skip to content

[Kernel] feat: add custom CUDA kernels for all sampling ops#1444

Open
AlpinDale wants to merge 4 commits into
mainfrom
topk-topp-kernels
Open

[Kernel] feat: add custom CUDA kernels for all sampling ops#1444
AlpinDale wants to merge 4 commits into
mainfrom
topk-topp-kernels

Conversation

@AlpinDale
Copy link
Copy Markdown
Collaborator

Trying to write custom kernels for every sampling op here.

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.

1 participant