transformer_engine/paddle/layer/attention.py · 161b1d98f80243c78ddecdadd15b010549e4e3d0 · OpenDAS / TransformerEngine

Shijie authored Jul 31, 2024



add deterministic option
Signed-off-by: Shijie Wang <jaywan@nvidia.com>
Co-authored-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

d793ca17

attention.py 44.1 KB