tests/pytorch/test_numerics.py · ec0d40d65d338e3015e63a75816454bfc7118155 · OpenDAS / TransformerEngine

PyTorch API numeric tests (#215) · ec0d40d6

Kirthi Shankar Sivamani authored May 11, 2023



* LayerNormMLP numeric test
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* DotProductAttention numeric test
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

---------
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

ec0d40d6

test_numerics.py 30.9 KB

Replace test_numerics.py