intel-extension-for-pytorch

1
import unittest
2
from common_utils import TestCase
3
import os
4
import subprocess
5

6

7
class TestConvReorder(TestCase):
8
    def test_conv_with_itensor_size1(self):
9
        num = 0
10
        loc = os.path.dirname(os.path.abspath(__file__))
11
        with subprocess.Popen(
12
            "DNNL_VERBOSE=1 python  -u {}/itensor_size1_test.py".format(loc),
13
            shell=True,
14
            stdout=subprocess.PIPE,
15
            stderr=subprocess.STDOUT,
16
        ) as p:
17
            for line in p.stdout.readlines():
18
                line = str(line, "utf-8").strip()
19
                if line.__contains__("reorder"):
20
                    num = num + 1
21
        assert num == 3, "conv channelslast has unexpected reorder."
22

23

24
if __name__ == "__main__":
25
    test = unittest.main()
26