xla
Add flash attention as custom op so dynamo can trace it
#6875

Merged

Add flash attention as custom op so dynamo can trace it #6875

alanwaketan merged 2 commits into master from JackCaoG/flash_attention_custom_op

Add flash attention as custom op so dynamo can trace it

41811947

JackCaoG added pallas

JackCaoG requested a review from

alanwaketan 2 years ago

linter

855836fa

JackCaoG commented on 2024-04-02

alanwaketan approved these changes on 2024-04-03

alanwaketan merged d9386809 into master 2 years ago

Reviewers

alanwaketan

Assignees

No one assigned

Labels

pallas

Milestone

No milestone