Skip to content

Conversation

@guocuimi
Copy link
Collaborator

@guocuimi guocuimi commented Jul 12, 2024

added kv_seq tile logical using cute::local_tile for testing purpose. similar logic will be used in cuda kernel later.

@guocuimi guocuimi changed the title kernel: added tile logic using cute::local_tile into cpu attention attention: added tile logic using cute::local_tile into cpu attention Jul 12, 2024
@guocuimi guocuimi merged commit f97014c into main Jul 12, 2024
@guocuimi guocuimi deleted the tile branch July 12, 2024 06:39
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