Skip to content

Flash Attention move to cute DSL Flash Attention #3083

@johnnynunez

Description

@johnnynunez

Ticket Type

🐛 Bug Report (Something isn't working)

Environment & System Info

FA2 and FA3 is now in cute dsl.
FA4 useful for jetson agx thor is in cute dsl too.

Move everything to cute dsl to avoid huge time in compilation.

Description

Flash attention C++ to FA cute DSL

Context & Reproduction

No response

Relevant logs or stack trace

Checklist

  • I have searched existing tickets to ensure this isn't a duplicate.
  • I am using the latest version of the main branch.
  • I have verified this is not an environment-specific problem.

Additional Info / Workarounds

No response

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn’t working correctly

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions