tests/L0/run_transformer/test_fused_softmax.py · 3ad9db2adb968c67dd509c5408eeca0884f6ab3f · OpenDAS / apex

Masaki Kozuki authored Oct 02, 2021


Co-authored-by: Piotr Bialecki <pbialecki@nvidia.com>
Co-authored-by: Eddie Yan <eddiey@nvidia.com>
Co-authored-by: Rishi Puri <riship@nvidia.com>
Co-authored-by: Sangkug Lym <slym@nvidia.com>

365fdc18

test_fused_softmax.py 6.14 KB

Replace test_fused_softmax.py