Skip to content

Commit 998ea37

Browse files
committed
Add test.
1 parent 49cd310 commit 998ea37

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

test/test_operations.py

+10
Original file line numberDiff line numberDiff line change
@@ -2384,6 +2384,16 @@ def test_cummax_0_sized_dimension(self):
23842384

23852385
self.assertEqual(actual, expected)
23862386

2387+
def test_conj_no_fallback(self):
2388+
met.clear_all()
2389+
2390+
tensor = torch.tensor([1 + 2j])
2391+
expected = torch.conj(tensor)
2392+
actual = torch.conj(tensor.to(xm.xla_device()))
2393+
2394+
self.assertEqual(actual, expected)
2395+
self.assertEqual(len(met.executed_fallback_ops()), 0)
2396+
23872397

23882398
class MNISTComparator(nn.Module):
23892399

0 commit comments

Comments
 (0)