tests/pytorch/test_sanity.py · d705f7ff7691675e31e31fda642f28691d9759e1 · 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_sanity.py 29.3 KB

Replace test_sanity.py