Jon Allen issues

Repositories
Issues
Comments

Results 2 issues of


                                            Jon Allen

Using system CUDA libraries

**Describe the bug** transformer-engine is currently searching for system CUDA binaries ( https://github.com/NVIDIA/TransformerEngine/blob/67fcc15255248a26be124de3854a47f84102f285/transformer_engine/common/__init__.py#L237). This is in conflict with Pytorch, which uses the CUDA Python packages (https://pypi.org/project/nvidia-cudnn-cu12/). **Steps/Code to reproduce bug**...

bug

Please add a license

Can you please add a license so that other people can extend and modify the code?