tinygrad/tinygrad
2
0
Fork 3
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-08-29 10:16:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

add test for Tensor silu and swish (#8381)

Browse Source 
was only tested in onnx, added to test_ops for completeness
This commit is contained in:
chenyu
2024-12-22 21:08:59 -05:00
committed by GitHub
parent 572ebd9f27
commit a556adf028
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/test_ops.py
+6
View File
@@ -765,6 +765,12 @@ class TestOps(unittest.TestCase):
def test_selu(self):
helper_test_op([(45,65)], torch.nn.functional.selu, Tensor.selu)
helper_test_op([()], torch.nn.functional.selu, Tensor.selu)
def test_silu(self):
helper_test_op([(45,65)], torch.nn.functional.silu, Tensor.silu)
helper_test_op([()], torch.nn.functional.silu, Tensor.silu)
def test_swish(self):
helper_test_op([(45,65)], torch.nn.functional.silu, Tensor.swish)
helper_test_op([()], torch.nn.functional.silu, Tensor.swish)
def test_abs(self):
helper_test_op([(45,65)], torch.abs, Tensor.abs)