intel-extension-for-pytorch

1
import torch
2
import unittest
3
from common_utils import TestCase
4

5

6
class TestTesorMethod(TestCase):
7
    def test_numpy(self):
8
        # float tensor, numpy array will share memory with torch tensor.
9
        x = torch.randn(2, 3)
10
        y = torch.from_numpy(x.numpy())
11
        self.assertEqual(x, y)
12
        self.assertEqual(x.data_ptr(), y.data_ptr())
13
        # bfloat16 tensor, numpy array will not share memory with torch tensor.
14
        x = torch.randn(2, 3).bfloat16()
15
        y = torch.from_numpy(x.numpy())
16
        self.assertEqual(x, y.bfloat16())
17
        self.assertNotEqual(x.data_ptr(), y.data_ptr())
18

19

20
if __name__ == "__main__":
21
    test = unittest.main()
22