(self)
| 27 | @gpu_test |
| 28 | @p100_exempt |
| 29 | def test_gpu_computation(self): |
| 30 | cuda = torch.device('cuda') |
| 31 | a = torch.tensor([1., 2.], device=cuda) |
| 32 | |
| 33 | result = a.sum() |
| 34 | |
| 35 | self.assertEqual(torch.tensor([3.], device=cuda), result) |
| 36 | |
| 37 | @gpu_test |
| 38 | @p100_exempt |
nothing calls this directly
no outgoing calls
no test coverage detected