(self)
| 18 | @gpu_test |
| 19 | def test_linalg(self): |
| 20 | A = torch.randn(3, 3).t().to('cuda') |
| 21 | B = torch.randn(3).t().to('cuda') |
| 22 | |
| 23 | result = torch.linalg.solve(A, B) |
| 24 | self.assertEqual(3, result.shape[0]) |
| 25 | |
| 26 | @gpu_test |
| 27 | def test_gpu_computation(self): |
| 28 | cuda = torch.device('cuda') |
nothing calls this directly
no outgoing calls
no test coverage detected