Skip to content

Question about MagiAttention requirement #4

Description

@dhcpack

Hi authors, thanks for the great work!

I noticed in the README that MagiAttention is required for grouped causal attention, and according to its documentation, it only supports Hopper and Blackwell GPUs. This hardware requirement seems quite restrictive.

I’m wondering whether MagiAttention can be replaced by other libraries (e.g., FlashAttention or similar implementations). Even if this leads to some efficiency degradation, it would greatly improve usability for users without access to newer GPUs.

Moreover, I noticed that in the supplementary material, the experiments were conducted on A800 GPUs. I’m curious whether the authors have any workaround or internal solution that allows MagiAttention to run on A100/A800 GPUs.

Any clarification would be greatly appreciated. Thanks again for your excellent work!

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions