mirror of
https://github.com/pytorch/pytorch.git
synced 2025-11-04 16:04:58 +08:00
02715d0876
Pull Request resolved: https://github.com/pytorch/pytorch/pull/157639 Approved by: https://github.com/yewentao256, https://github.com/jansel ghstack dependencies: #157638
83 lines
2.3 KiB
Python
83 lines
2.3 KiB
Python
# Owner(s): ["module: dynamo"]
|
|
import unittest
|
|
|
|
from torch._dynamo import config
|
|
from torch._dynamo.testing import make_test_cls_with_patches
|
|
|
|
|
|
try:
|
|
from . import test_export
|
|
except ImportError:
|
|
import test_export
|
|
|
|
|
|
test_classes = {}
|
|
|
|
|
|
def make_dynamic_cls(cls):
|
|
suffix = "_inline_and_install"
|
|
|
|
cls_prefix = "InlineAndInstall"
|
|
|
|
test_class = make_test_cls_with_patches(
|
|
cls,
|
|
cls_prefix,
|
|
suffix,
|
|
(config, "install_free_tensors", True),
|
|
(config, "inline_inbuilt_nn_modules", True),
|
|
xfail_prop="_expected_failure_inline_and_install",
|
|
)
|
|
|
|
test_classes[test_class.__name__] = test_class
|
|
# REMOVING THIS LINE WILL STOP TESTS FROM RUNNING
|
|
globals()[test_class.__name__] = test_class
|
|
test_class.__module__ = __name__
|
|
return test_class
|
|
|
|
|
|
tests = [
|
|
test_export.ExportTests,
|
|
]
|
|
for test in tests:
|
|
make_dynamic_cls(test)
|
|
del test
|
|
|
|
# After installing and inlining is turned on, these tests won't throw
|
|
# errors in export (which is expected for the test to pass)
|
|
# Therefore, these unittest are expected to fail, and we need to update the
|
|
# semantics
|
|
unittest.expectedFailure(
|
|
InlineAndInstallExportTests.test_invalid_input_global_inline_and_install # noqa: F821
|
|
)
|
|
unittest.expectedFailure(
|
|
InlineAndInstallExportTests.test_invalid_input_global_multiple_access_inline_and_install # noqa: F821
|
|
)
|
|
unittest.expectedFailure(
|
|
InlineAndInstallExportTests.test_invalid_input_nonlocal_inline_and_install # noqa: F821
|
|
)
|
|
|
|
|
|
# These tests do string comparison on the graphs, and since buffers are now inlined, they
|
|
# are named different, resulting in failure
|
|
unittest.expectedFailure(
|
|
InlineAndInstallExportTests.test_param_buffer_safe_from_mutation_simple_inline_and_install # noqa: F821
|
|
)
|
|
|
|
|
|
# This particular test is marked expecting failure, since dynamo was creating second param for a
|
|
# and this was causing a failure in the sum; however with these changes, that test is fixed
|
|
# so will now pass, so we need to mark that it is no longer expected to fail
|
|
def expectedSuccess(test_item):
|
|
test_item.__unittest_expecting_failure__ = False
|
|
return test_item
|
|
|
|
|
|
expectedSuccess(
|
|
InlineAndInstallExportTests.test_sum_param_inline_and_install # noqa: F821
|
|
)
|
|
|
|
if __name__ == "__main__":
|
|
from torch._dynamo.test_case import run_tests
|
|
|
|
run_tests()
|