Skip to content

logit_bias support for NEW_UNSHARDED_ATTN_KERNEL#24

Open
HahTK wants to merge 6 commits intolow_tpfrom
low_tp_lr
Open

logit_bias support for NEW_UNSHARDED_ATTN_KERNEL#24
HahTK wants to merge 6 commits intolow_tpfrom
low_tp_lr

Conversation

@HahTK
Copy link
Copy Markdown
Collaborator

@HahTK HahTK commented Dec 4, 2024

Add logit_bias support to ENABLE_NEW_UNSHARDED_ATTN_KERNEL. Old legacy kernel does not support this. Additionally, changes here needs to be merged with upstream which has significant changes (example, sharding and 2D logit biases)

Also fixed assertion typo.

@HahTK HahTK assigned HahTK, indhub and apoorvtintin and unassigned HahTK Dec 5, 2024
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