intel-extension-for-pytorch

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

6

7
class TestProfiler(TestCase):
8
    # currently only check ipex softmax as an example
9
    def test_profile_on(self):
10
        num = 0
11
        loc = os.path.dirname(os.path.abspath(__file__))
12
        with subprocess.Popen(
13
            "python  -u {}/profile_ipex_op.py".format(loc),
14
            shell=True,
15
            stdout=subprocess.PIPE,
16
            stderr=subprocess.STDOUT,
17
        ) as p:
18
            for line in p.stdout.readlines():
19
                line = str(line, "utf-8").strip()
20
                if line.__contains__("dil_softmax_"):
21
                    num = num + 1
22
        assert num == 2, "IPEX op profiling info not found."
23

24

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