transformer_engine/pytorch/attention.py · bacefdbb6815159c42d5ca501a1400697b98a1e3 · 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