-
-
Notifications
You must be signed in to change notification settings - Fork 4.7k
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.
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
Add TransformerEngine #26296
base: main
Are you sure you want to change the base?
Add TransformerEngine #26296
Conversation
Hi! This is the friendly automated conda-forge-linting service. I wanted to let you know that I linted all conda-recipes in your PR ( Here's what I've got... For TransformerEngine:
|
A library for accelerating Transformer models on NVIDIA GPUs! Repo at https://github.com/NVIDIA/TransformerEngine
bd391e7
to
9bad097
Compare
Hi! This is the friendly automated conda-forge-linting service. I just wanted to let you know that I linted all conda-recipes in your PR ( |
- pip | ||
- python | ||
run: | ||
- flash-attn >=2.0.6,<=2.4.2,!=2.0.9,!=2.1.0 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Wait for NVIDIA/TransformerEngine#838 and a subsequent release, since conda-forge only has flash-attn=2.5.8
at the moment.
Bumps [TransformerEngine](https://github.com/NVIDIA/TransformerEngine) from 1.5 to 1.6. - [Release notes](https://docs.nvidia.com/deeplearning/transformer-engine/release-notes/index.html) - [Commits](NVIDIA/TransformerEngine@v1.5...v1.6)
So TransformerEngine v1.6 has a git submodule dependency on cudnn-frontend v1.3.0 (see https://github.com/NVIDIA/TransformerEngine/tree/v1.6/3rdparty). From reading conda-forge/conda-forge.github.io#619, it sounds like the proper way is to actually have a |
A library for accelerating Transformer models on NVIDIA GPUs! Repo at https://github.com/NVIDIA/TransformerEngine
Checklist
url
) rather than a repo (e.g.git_url
) is used in your recipe (see here for more details).