David Berard cead0363a8 [jit][nested strided tensor] support nested tensor in check_trace (#121039) ...

Summary:
torch.testing.assert_equal doesn't support nested strided tensors because sizes is not implemented.

This adds special handling for nested tensors by checking for nested tensors unbinding if they are found.

Test Plan: test_trace_with_nested_strided_tensor_output

Differential Revision: D54430238

Pull Request resolved: https://github.com/pytorch/pytorch/pull/121039
Approved by: https://github.com/YuqingJ

2024-03-04 01:15:45 +00:00

..

_imported_class_test

[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308)

2023-02-07 21:10:56 +00:00

fixtures

update test fixture (#89796)

2023-02-07 14:58:57 +00:00

fixtures_srcs

[BE]: Update Ruff to 0.0.280 (#105724)

2023-07-22 23:03:34 +00:00

xnnpack

[BE] [3/3] Rewrite super() calls in test (#94592)

2023-02-12 22:20:53 +00:00

__init__.py

…

myexception.py

[BE] [3/3] Rewrite super() calls in test (#94592)

2023-02-12 22:20:53 +00:00

test_alias_analysis.py

…

test_async.py

[BE]: Enable F821 and fix bugs (#116579)

2024-01-01 08:40:46 +00:00

test_aten_pow.py

…

test_attr.py

[BE] [3/3] Rewrite super() calls in test (#94592)

2023-02-12 22:20:53 +00:00

test_autodiff_subgraph_slicing.py

[BE]: Update flake8 to v6.1.0 and fix lints (#116591)

2024-01-03 06:04:44 +00:00

test_autodiff.py

PT2/TorchScript interoperability fix (#94678)

2023-02-15 01:21:10 +00:00

test_await.py

[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308)

2023-02-07 21:10:56 +00:00

test_backend_nnapi.py

Run test_jit with PYTORCH_TEST_WITH_DYNAMO=1 in CI (#117765)

2024-01-19 13:42:41 +00:00

test_backends.py

[BE]: Enable F821 and fix bugs (#116579)

2024-01-01 08:40:46 +00:00

test_batch_mm.py

…

test_builtins.py

[BE]: Enable F821 and fix bugs (#116579)

2024-01-01 08:40:46 +00:00

test_class_type.py

[BE] [3/3] Rewrite super() calls in test (#94592)

2023-02-12 22:20:53 +00:00

test_complex.py

…

test_complexity.py

[jit] test_complexity.py - don't set default dtype in global scope (#106486)

2023-08-03 02:50:15 +00:00

test_convert_activation.py

[BE] [3/3] Rewrite super() calls in test (#94592)

2023-02-12 22:20:53 +00:00

test_cuda.py

Refactor some tests by using TEST_CUDA & TEST_MULTIGPU instead (#116083)

2024-01-03 08:53:59 +00:00

test_custom_operators.py

Raise AttributeError in _OpsNamespace if __self__ attribute is requested (#104096)

2023-06-27 01:42:06 +00:00

test_data_parallel.py

…

test_dataclasses.py

Set min supported Python version to 3.8 (#93155)

2023-01-29 18:28:46 +00:00

test_dce.py

[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308)

2023-02-07 21:10:56 +00:00

test_device_analysis.py

…

test_dtype_analysis.py

[BE]: Apply ruff PERF fixes to torch (#104917)

2023-07-11 20:45:21 +00:00

test_enum.py

[JIT] Fix typed enum handling in 3.11 (#109717)

2023-09-20 22:09:41 +00:00

test_exception.py

[BE]: Enable F821 and fix bugs (#116579)

2024-01-01 08:40:46 +00:00

test_freezing.py

Linear-BN Fusion: add precondition check (#119264)

2024-02-13 04:16:34 +00:00

test_functional_blocks.py

…

test_fuser_common.py

…

test_generator.py

Add support for torch.Generator type in TorchScript (#110413)

2023-11-21 23:07:21 +00:00

test_graph_rewrite_passes.py

[BE] [3/3] Rewrite super() calls in test (#94592)

2023-02-12 22:20:53 +00:00

test_hash.py

[Reland] Add py-3.10 config (#82329)

2022-07-27 20:22:47 +00:00

test_hooks_modules.py

…

test_hooks.py

Fix typos in messages under test (#89121)

2022-11-17 01:55:03 +00:00

test_ignorable_args.py

…

test_ignore_context_manager.py

[BE] [3/3] Rewrite super() calls in test (#94592)

2023-02-12 22:20:53 +00:00

test_isinstance.py

…

test_jit_utils.py

jit trace will fail for parameter check if it contains param whose ki… (#94032)

2023-02-13 20:33:30 +00:00

test_list_dict.py

Add linux cpu test for 3.12 (#117853)

2024-02-14 20:52:23 +00:00

test_logging.py

…

test_misc.py

[BE]: Enable F821 and fix bugs (#116579)

2024-01-01 08:40:46 +00:00

test_models.py

[quant] Remove deprecated torch.jit.quantized APIs (#118406)

2024-01-27 18:32:45 +00:00

test_module_apis.py

…

test_module_containers.py

[JIT] clarify errors due to non-literal indexing into ModuleList, ModuleDict (#98606)

2023-04-18 02:53:53 +00:00

test_module_interface.py

[BE] [3/3] Rewrite super() calls in test (#94592)

2023-02-12 22:20:53 +00:00

test_modules.py

…

test_op_decompositions.py

…

test_optimize_for_mobile_preserve_debug_info.py

[BE] [3/3] Rewrite super() calls in test (#94592)

2023-02-12 22:20:53 +00:00

test_parametrization.py

Use generators with all/any in torch/optim (#78142)

2022-06-24 17:23:45 +00:00

test_pdt.py

Reduce pytest blocklist (#96016)

2023-03-07 18:30:27 +00:00

test_peephole.py

[BE] [3/3] Rewrite super() calls in test (#94592)

2023-02-12 22:20:53 +00:00

test_profiler.py

PT2/TorchScript interoperability fix (#94678)

2023-02-15 01:21:10 +00:00

test_python_bindings.py

Fix Python-bound function signature (torch._C.Graph.addInput) (#88528)

2022-11-09 01:31:45 +00:00

test_python_builtins.py

…

test_python_ir.py

[JIT] python IR bindings: consolidate tests, add short docs in OVERVIEW.md (#118319)

2024-01-27 03:11:51 +00:00

test_recursive_script.py

[BE]: Enable F821 and fix bugs (#116579)

2024-01-01 08:40:46 +00:00

test_remove_mutation.py

Remove set_default_dtype calls from jit and ops tests (#105072)

2023-07-15 03:18:33 +00:00

test_save_load_for_op_version.py

[BE] [3/3] Rewrite super() calls in test (#94592)

2023-02-12 22:20:53 +00:00

test_save_load.py

[codemod] markDynamoStrictTest batch 22 (#117729)

2024-01-18 16:59:26 +00:00

test_script_profile.py

[BE] [3/3] Rewrite super() calls in test (#94592)

2023-02-12 22:20:53 +00:00

test_scriptmod_ann.py

[BE]: Update flake8 to v6.1.0 and fix lints (#116591)

2024-01-03 06:04:44 +00:00

test_slice.py

[BE] Apply almost all remaining flake8-comprehension checks (#94676)

2023-02-12 01:01:25 +00:00

test_sparse.py

…

test_string_formatting.py

…

test_symbolic_shape_analysis.py

[BE]: Enable F821 and fix bugs (#116579)

2024-01-01 08:40:46 +00:00

test_tensor_creation_ops.py

[primTorch] Refs for pdist, triu and related ops (#82819)

2022-08-18 21:52:03 +00:00

test_tensor_methods.py

…

test_torchbind.py

Run test_jit with PYTORCH_TEST_WITH_DYNAMO=1 in CI (#117765)

2024-01-19 13:42:41 +00:00

test_tracer.py

[jit][nested strided tensor] support nested tensor in check_trace (#121039)

2024-03-04 01:15:45 +00:00

test_type_sharing.py

[BE] [3/3] Rewrite super() calls in test (#94592)

2023-02-12 22:20:53 +00:00

test_types.py

[BE] [3/3] Rewrite super() calls in test (#94592)

2023-02-12 22:20:53 +00:00

test_typing.py

[BE]: Enable F821 and fix bugs (#116579)

2024-01-01 08:40:46 +00:00

test_union_pep604.py

Python 3.10 Union operator | support for JIT (#109293)

2023-09-25 15:35:54 +00:00

test_union.py

[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308)

2023-02-07 21:10:56 +00:00

test_unsupported_ops.py

…

test_upgraders.py

Get rid of ENABLE_UPGRADERS macro (#77574)

2022-08-09 05:33:14 +00:00

test_warn.py

…

test_with.py

[BE]: enable PLE error codes in ruff and fix bugs (#101079)

2023-05-11 23:57:25 +00:00