tests/pytorch/test_numerics.py · 53a3bc354141a721337a852093a8c909bee6a239 · OpenDAS / TransformerEngine

"vscode:/vscode.git/clone" did not exist on "a576a279629b4604a814ba8331d9a99ac6848c4c"

[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