transformer_engine/pytorch/attention.py · a326e351a1fb9c4ff8ee970a407c1f4f35f663af · OpenDAS / TransformerEngine

Marks101 authored Aug 15, 2024


Signed-off-by: Markus Schnoes <markus.schnoes@gmx.de>
Co-authored-by: Charlene Yang <8636796+cyanguwa@users.noreply.github.com>

a326e351

attention.py 294 KB