tests/pytorch/test_fused_attn.py · d097883e4618fcbdbab980d03a235e256435d3b0 · OpenDAS / TransformerEngine

Better way of checking cuDNN version (#485) · d097883e

Przemyslaw Tredak authored Oct 20, 2023



* Ability to check cuDNN version from Python
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>

* Modify the fused attention test to not use the CUDNN_VERSION env
variable which is specific to NGC containers
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>

---------
Signed-off-by: Przemek Tredak <ptredak@nvidia.com>

d097883e

test_fused_attn.py 36.4 KB

Replace test_fused_attn.py