transformer_engine/pytorch/attention.py · 4dc36f0e2959219b11180c156a32161db7afa43b · OpenDAS / TransformerEngine

Fix failing CI due to PR #557 merge (#616) · bacefdbb

Sudhakar Singh authored Jan 19, 2024



fix failing tests due to PR #557
Signed-off-by: Sudhakar Singh <sudhakars@nvidia.com>
Co-authored-by: cyanguwa <8636796+cyanguwa@users.noreply.github.com>

bacefdbb

attention.py 159 KB

Replace attention.py