test/dynamo/test_dynamic_shapes.py

# Owner(s): ["module: dynamo"]

from torch._dynamo.testing import make_test_cls_with_patches

try:
    from . import (
        test_export,
        test_functions,
        test_misc,
        test_modules,
        test_repros,
        test_subgraphs,
        test_unspec,
    )
except ImportError:
    import test_export
    import test_functions
    import test_misc
    import test_modules
    import test_repros
    import test_subgraphs
    import test_unspec

import unittest


def make_dynamic_cls(cls):
    return make_test_cls_with_patches(
        cls, "DynamicShapes", "_dynamic_shapes", ("dynamic_shapes", True)
    )


DynamicShapesFunctionTests = make_dynamic_cls(test_functions.FunctionTests)
DynamicShapesMiscTests = make_dynamic_cls(test_misc.MiscTests)
DynamicShapesReproTests = make_dynamic_cls(test_repros.ReproTests)
DynamicShapesNNModuleTests = make_dynamic_cls(test_modules.NNModuleTests)
DynamicShapesUnspecTests = make_dynamic_cls(test_unspec.UnspecTests)
DynamicShapesExportTests = make_dynamic_cls(test_export.ExportTests)
DynamicShapesSubGraphTests = make_dynamic_cls(test_subgraphs.SubGraphTests)


# DynamicShapesFunctionTests
unittest.expectedFailure(
    DynamicShapesFunctionTests.test_len_tensor_dynamic_shapes
    # TypeError: 'torch._C.SymIntNode' object cannot be interpreted as an integer
)

unittest.expectedFailure(
    DynamicShapesFunctionTests.test_tensor_len_dynamic_shapes
    # TypeError: 'torch._C.SymIntNode' object cannot be interpreted as an integer
)


unittest.expectedFailure(
    DynamicShapesReproTests.test_do_paste_mask_dynamic_shapes
    # aten.min.dim - couldn't find symbolic meta function/decomposition
)

unittest.expectedFailure(
    DynamicShapesReproTests.test_convert_boxes_to_pooler_format_dynamic_shapes
    # Could not infer dtype of torch._C.SymIntNode
)

unittest.expectedFailure(
    DynamicShapesReproTests.test_hf_t5_forward_dynamic_shapes
    # Cannot call sizes() on tensor with symbolic sizes/strides
)

# DynamicShapesExportTests
unittest.expectedFailure(
    DynamicShapesExportTests.test_export_with_constant_list_nonzero_dynamic_shapes
)
unittest.expectedFailure(
    DynamicShapesExportTests.test_export_with_constant_list_nonzero_free_function_dynamic_shapes
)
unittest.expectedFailure(
    DynamicShapesExportTests.test_export_with_constant_tuple_nonzero_dynamic_shapes
)
unittest.expectedFailure(
    DynamicShapesExportTests.test_export_with_constant_tuple_nonzero_dynamic_shapes
)


# DynamicShapesSubGraphTests
unittest.expectedFailure(
    DynamicShapesSubGraphTests.test_enumerate_not_break_graph_dynamic_shapes
)
unittest.expectedFailure(DynamicShapesSubGraphTests.test_restore_state_dynamic_shapes)

# DynamicShapesUnspecTests
# Missing decomp
# RuntimeError: Failed running call_function <function batch_norm at 0x7f7d1ce38310>
# (*(FakeTensor(FakeTensor(..., device='meta', size=(5, 1, 28, 28)), cpu),
# FakeTensor(FakeTensor(..., device='meta', size=(1,)), cpu),
#  FakeTensor(FakeTensor(..., device='meta', size=(1,)), cpu),
#  FakeTensor(Parameter(FakeTensor(..., device='meta', size=(1,),
#  requires_grad=True)), cpu),
#  FakeTensor(Parameter(FakeTensor(..., device='meta', size=(1,),
#  requires_grad=True)), cpu), False, 0.1,
# FakeTensor(FakeTensor(..., device='meta', size=()), cpu)), **{}):
# aten._local_scalar_dense.default
unittest.expectedFailure(test_unspec.UnspecReproTests.test_batch_norm_act_unspec)

# SymIntArrayRef expected to contain only concrete integers
unittest.expectedFailure(
    DynamicShapesUnspecTests.test_unspec_float_precision_dynamic_shapes
)

unittest.expectedFailure(
    DynamicShapesReproTests.test_reformer_sorting_dynamic_shapes
    # Unable to cast Python instance to C++ type
)

if __name__ == "__main__":
    from torch._dynamo.test_case import run_tests

    run_tests()