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

Commit

2 years ago

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

References

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

Author

JackCaoG

JackCaoG

Parents

Loading