New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[Bug] SUPPORT_FLASH1 = digit_version(torch.version) >= digit_version('2… #587

Merged

pppppM merged 2 commits into InternLM:main from HIT-cwh:fix_sdpa_version

Apr 19, 2024

Collaborator

HIT-cwh commented Apr 19, 2024

….1.1')

HIT-cwh added 2 commits

April 19, 2024 14:17


          SUPPORT_FLASH1 = digit_version(torch.__version__) >= digit_version('2…

677e4ce

….1.1')


          add comments why SUPPORT_FLASH1 = digit_version(torch.__version__) >=…

ae35f85

… digit_version('2.1.1')

pppppM approved these changes

View reviewed changes

pppppM merged commit 6ad3268 into InternLM:main

1 of 3 checks passed

HIT-cwh deleted the fix_sdpa_version branch

April 25, 2024 06:33

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