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

[Pytorch] Added squared ReLU implementation (#846) · 53a3bc35

Phuong Nguyen authored May 16, 2024



* added squared relu in te-torch
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>

---------
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>

53a3bc35

test_numerics.py 49.7 KB

Replace test_numerics.py