Replies: 1 comment
-
I witnessed that the kernel would output would someone be able to point out the issue? Thank you!
|
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
I would like to add a custom mask of shape
(batch_size, head_dim, seq_length, seq_length)
inside the kernel. I am not sure how to find the appropriate range of pointers to the mask inside the for loop. Can anyone offer some insights on this? Thank you!Beta Was this translation helpful? Give feedback.
All reactions