xla
2298cbcb - Add flash attention as custom op so dynamo can trace it (#6875)

Commit

1 year ago

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

References

#6884 - [Backport] Add flash attention as custom op so dynamo can trace it (#6875)

Author

JackCaoG

JackCaoG

Committer

JackCaoG

JackCaoG

Parents

Loading