More tests

pytorch-labs · Jan 6, 2025 · f72111a · f72111a
1 parent d70fc51
commit f72111a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/fixtures/misc/checker/logsumexp.py b/tests/fixtures/misc/checker/logsumexp.py
@@ -9,3 +9,6 @@
 # not logsumexp
 y = torch.log(torch.sum(torch.exp(x), 1, keepdim=True) + 2.5)
 y = torch.log(torch.sum(torch.exp(x) + 2.5, 1))
+y = torch.log(2 + x)
+y = torch.sum(torch.log(torch.exp(x)), 1)
+y = torch.exp(torch.sum(torch.log(x), 1, keepdim=True))