xla
41811947 - Add flash attention as custom op so dynamo can trace it

Commit

1 year ago

Add flash attention as custom op so dynamo can trace it

References

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

Author

JackCaoG

JackCaoG

Parents

Loading