mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
fbe6f42dcf
See https://github.com/pytorch/pytorch/pull/129751#issue-2380881501. Most changes are auto-generated by linter. You can review these PRs via: ```bash git diff --ignore-all-space --ignore-blank-lines HEAD~1 ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/129759 Approved by: https://github.com/justinchuby, https://github.com/ezyang
44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
# Owner(s): ["oncall: mobile"]
|
|
|
|
import os
|
|
import tempfile
|
|
from pathlib import Path
|
|
|
|
from torch.jit.generate_bytecode import generate_upgraders_bytecode
|
|
from torch.testing._internal.common_utils import run_tests, TestCase
|
|
from torchgen.operator_versions.gen_mobile_upgraders import sort_upgrader, write_cpp
|
|
|
|
|
|
pytorch_caffe2_dir = Path(__file__).resolve().parents[2]
|
|
|
|
|
|
class TestLiteScriptModule(TestCase):
|
|
def test_generate_bytecode(self):
|
|
upgrader_list = generate_upgraders_bytecode()
|
|
sorted_upgrader_list = sort_upgrader(upgrader_list)
|
|
upgrader_mobile_cpp_path = (
|
|
pytorch_caffe2_dir
|
|
/ "torch"
|
|
/ "csrc"
|
|
/ "jit"
|
|
/ "mobile"
|
|
/ "upgrader_mobile.cpp"
|
|
)
|
|
with tempfile.TemporaryDirectory() as tmpdirname:
|
|
write_cpp(tmpdirname, sorted_upgrader_list)
|
|
with open(os.path.join(tmpdirname, "upgrader_mobile.cpp")) as file_name:
|
|
actual_output = [line.strip() for line in file_name if line]
|
|
with open(str(upgrader_mobile_cpp_path)) as file_name:
|
|
expect_output = [line.strip() for line in file_name if line]
|
|
actual_output_filtered = list(
|
|
filter(lambda token: len(token) != 0, actual_output)
|
|
)
|
|
expect_output_filtered = list(
|
|
filter(lambda token: len(token) != 0, expect_output)
|
|
)
|
|
|
|
self.assertEqual(actual_output_filtered, expect_output_filtered)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run_tests()
|