transformer_engine/cmake/FindCUDNN.cmake · 90c267f2748ef4b4cee2ef3ef3068f5e91ed645b · OpenDAS / TransformerEngine

Update FindCUDNN.cmake for cuDNN 9 (#640) · e2803b16

cyanguwa authored Jan 31, 2024



* update cudnn cmake for v9
Signed-off-by: Charlene Yang <8636796+cyanguwa@users.noreply.github.com>

* add back license information
Signed-off-by: Charlene Yang <8636796+cyanguwa@users.noreply.github.com>

---------
Signed-off-by: Charlene Yang <8636796+cyanguwa@users.noreply.github.com>

e2803b16

FindCUDNN.cmake 2.99 KB

Replace FindCUDNN.cmake