Closed timmoon10 closed 1 week ago
The _cudnn_version function was removed from the fused attention tests in https://github.com/NVIDIA/TransformerEngine/pull/889, but not from the corresponding tests for context parallelism. This has already been fixed in the main branch with https://github.com/NVIDIA/TransformerEngine/pull/927, but not in the 1.8 release.
_cudnn_version
main
Description
The
_cudnn_version
function was removed from the fused attention tests in https://github.com/NVIDIA/TransformerEngine/pull/889, but not from the corresponding tests for context parallelism. This has already been fixed in themain
branch with https://github.com/NVIDIA/TransformerEngine/pull/927, but not in the 1.8 release.Type of change
Changes
Checklist: