pytorch/test
Wei-Sheng Chin 357a06f7c9 [ONNX] Fix type promotion pass (#118246)
Currently, when `node.meta["val"]` is `torch.Sym*`, its `hint` [is extracted](61865205b6/torch/onnx/_internal/fx/passes/type_promotion.py (L86)) and used in type promotion. However, it will [override](61865205b6/torch/onnx/_internal/fx/passes/type_promotion.py (L1409)) dynamic shape information carried in `node.meta["val"]` during [type propagation](61865205b6/torch/onnx/_internal/fx/passes/type_promotion.py (L1401)) and the FX graph seen in `onnxrt` always carries static shapes. Let's use `torch.Sym*` directly so that the type promotion propagates and stores dynamic shapes.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/118246
Approved by: https://github.com/titaiwangms
2024-01-25 07:04:18 +00:00
..
ao/sparsity [BE]: Update flake8 to v6.1.0 and fix lints (#116591) 2024-01-03 06:04:44 +00:00
autograd Autograd attaches logging hooks only in debug level (#116522) 2024-01-02 20:06:18 +00:00
backends/xeon
benchmark_utils
bottleneck_test
cpp Replace constraints with dynamic_shapes in caffe2/test/cpp & torchrec/distributed/tests/test_pt2 (#118026) 2024-01-23 22:15:15 +00:00
cpp_api_parity
cpp_extensions Add function to materialize COW storages (#117053) 2024-01-10 15:34:16 +00:00
custom_backend
custom_operator Better unsupported op error message (#117770) 2024-01-23 15:05:16 +00:00
distributed [DTensor][BE] re-enable test_dtensor_ops in CPU CI (#118134) 2024-01-24 22:11:51 +00:00
distributions Add bfloat16 CUDA support to binomial distribution (#116932) 2024-01-07 19:50:10 +00:00
dynamo Fix Auto Functionalize to handle specified default values (#118035) 2024-01-25 01:22:12 +00:00
edge
error_messages
expect Revert "[CUDNN][SDPA] Experimental cuDNN Flash Attention v2 Inference (#115663)" 2024-01-18 23:40:30 +00:00
export [fx] Add hooks to intercept node replacements. (#117825) 2024-01-23 22:28:40 +00:00
forward_backward_compatibility Add out_dtype support for sparse semi-structured CUTLASS back-end (#116519) 2024-01-03 16:23:17 +00:00
functorch [ROCm] skip test_eager_transforms.py test_compile_vmap_hessian_cuda (#118009) 2024-01-23 09:57:18 +00:00
fx [BE]: Update flake8 to v6.1.0 and fix lints (#116591) 2024-01-03 06:04:44 +00:00
inductor Log stack trace of mutated idx reland (#118110) 2024-01-25 00:30:03 +00:00
jit Run test_jit with PYTORCH_TEST_WITH_DYNAMO=1 in CI (#117765) 2024-01-19 13:42:41 +00:00
jit_hooks
lazy Replaced CHECK with TORCH_CHECK in order to not abort, but throw a Ru… (#117653) 2024-01-18 07:47:22 +00:00
mobile [BE]: Update flake8 to v6.1.0 and fix lints (#116591) 2024-01-03 06:04:44 +00:00
nn [BE]: Update flake8 to v6.1.0 and fix lints (#116591) 2024-01-03 06:04:44 +00:00
onnx [ONNX] Fix type promotion pass (#118246) 2024-01-25 07:04:18 +00:00
onnx_caffe2
optim Migrate test_complex_optimizer to OptimizerInfo (#118160) 2024-01-24 21:22:47 +00:00
package
profiler Revert "accelerate binary_cross_entropy_with_logits by using log_sigmoid operator (#115539)" 2024-01-22 14:48:35 +00:00
quantization [quant][pt2e] Add fold_quantize=True for all convert_pt2e calls (#117797) 2024-01-24 17:54:13 +00:00
scripts
test_img
torch_np Add super().setUp() to TestFFT1D (#117329) 2024-01-12 07:47:01 +00:00
typing
_test_bazel.py
allowlist_for_publicAPI.json Initial torchbind support in PT2 (#117697) 2024-01-19 06:28:20 +00:00
conftest.py Reduce pytest prints (#117069) 2024-01-23 18:39:30 +00:00
create_dummy_torchscript_model.py
delete.py
HowToWriteTestsUsingFileCheck.md
linear.py
load_torchscript_model.py
minioptest_failures_dict.json
mkl_verbose.py
mkldnn_verbose.py
pytest_shard_custom.py Reduce pytest prints (#117069) 2024-01-23 18:39:30 +00:00
run_doctests.sh
run_test.py Reduce pytest prints (#117069) 2024-01-23 18:39:30 +00:00
simulate_nccl_errors.py
test_ao_sparsity.py
test_autocast.py Remove incorrect usages of skipIfTorchDynamo (#117114) 2024-01-10 22:25:31 +00:00
test_autograd.py Support custom autograd Function forward AD return non-Tensor in forward (#118234) 2024-01-25 03:24:29 +00:00
test_autograd_fallback.py
test_binary_ufuncs.py [Dynamo] Add complex to literal constant (#117819) 2024-01-23 23:46:46 +00:00
test_bundled_images.py [BE]: Update flake8 to v6.1.0 and fix lints (#116591) 2024-01-03 06:04:44 +00:00
test_bundled_inputs.py
test_comparison_utils.py
test_compile_benchmark_util.py
test_complex.py
test_content_store.py
test_cpp_api_parity.py markDynamoStrictTest some test_cpp_extensions.* (#115858) 2023-12-15 01:22:38 +00:00
test_cpp_extensions_aot.py markDynamoStrictTest some test_cpp_extensions.* (#115858) 2023-12-15 01:22:38 +00:00
test_cpp_extensions_jit.py markDynamoStrictTest some test_cpp_extensions.* (#115858) 2023-12-15 01:22:38 +00:00
test_cpp_extensions_open_device_registration.py markDynamoStrictTest some test_cpp_extensions.* (#115858) 2023-12-15 01:22:38 +00:00
test_cuda.py [Memory Snapshot] Track context for SEGMENT_FREE and SEGMENT_UNMAP (#118055) 2024-01-23 21:48:57 +00:00
test_cuda_expandable_segments.py
test_cuda_multigpu.py
test_cuda_nvml_based_avail.py markDynamoStrictTest some more tests (#115857) 2023-12-15 01:22:38 +00:00
test_cuda_primary_ctx.py markDynamoStrictTest some more tests (#115857) 2023-12-15 01:22:38 +00:00
test_cuda_sanitizer.py
test_cuda_trace.py markDynamoStrictTest some more tests (#115857) 2023-12-15 01:22:38 +00:00
test_custom_ops.py Add config toggle suggestions for data-dependent/dynamic output shape (#114337) 2024-01-05 14:01:01 +00:00
test_dataloader.py [CI] Install dill in ci (#116214) 2024-01-24 23:42:35 +00:00
test_datapipe.py [CI] Install dill in ci (#116214) 2024-01-24 23:42:35 +00:00
test_decomp.py [PyTorch] SDPA decomp: actually use attn_mask (#117579) 2024-01-17 10:26:43 +00:00
test_deploy.py
test_determination.py
test_dispatch.py Revert "markDynamoStrictTest more tests (#115871)" 2023-12-19 15:36:27 +00:00
test_dlpack.py Improve uint{16,32,64} dlpack/numpy compatibility (#116808) 2024-01-11 17:01:54 +00:00
test_dynamic_shapes.py Rename unbacked SymInt prefix to u (#117859) 2024-01-22 20:53:47 +00:00
test_expanded_weights.py [Dynamo] Use isinstance rather than istype when check if python module type (#117022) 2024-01-15 23:25:30 +00:00
test_fake_tensor.py Initial implementation of FakeTensor caching (#113873) 2024-01-17 20:38:54 +00:00
test_flop_counter.py Re-add initial Flash Attention support on ROCM (#115981) 2024-01-04 22:21:31 +00:00
test_foreach.py Refactor some tests by using TEST_CUDA & TEST_MULTIGPU instead (#116083) 2024-01-03 08:53:59 +00:00
test_function_schema.py
test_functional_autograd_benchmark.py
test_functional_optim.py
test_functionalization.py Enable reverse view_funcs by default for python subclasses (#116512) 2024-01-05 16:48:12 +00:00
test_functionalization_of_rng_ops.py
test_futures.py
test_fx.py [codemod] markDynamoStrictTest batch 22 (#117729) 2024-01-18 16:59:26 +00:00
test_fx_experimental.py Revert "[codemod] markDynamoStrictTest batch 16 (#117218)" 2024-01-12 03:06:20 +00:00
test_fx_passes.py
test_fx_reinplace_pass.py
test_hub.py
test_import_stats.py
test_indexing.py [BE]: Apply FURB118 (prev): replaces unnecessary lambdas with operator. (#116027) 2023-12-20 19:35:08 +00:00
test_itt.py
test_jit.py [BE]: Update flake8 to v6.1.0 and fix lints (#116591) 2024-01-03 06:04:44 +00:00
test_jit_autocast.py
test_jit_disabled.py
test_jit_fuser.py [BE]: Enable F821 and fix bugs (#116579) 2024-01-01 08:40:46 +00:00
test_jit_fuser_legacy.py
test_jit_fuser_te.py [BE]: Enable F821 and fix bugs (#116579) 2024-01-01 08:40:46 +00:00
test_jit_legacy.py
test_jit_llga_fuser.py
test_jit_profiling.py
test_jit_simple.py
test_jit_string.py
test_jiterator.py
test_kernel_launch_checks.py
test_legacy_vmap.py Skip some slow tests (under Dynamo) (#117389) 2024-01-12 22:18:07 +00:00
test_license.py
test_linalg.py enable mkldnn bf32 matmul (#116015) 2024-01-20 09:30:23 +00:00
test_logging.py
test_masked.py
test_maskedtensor.py
test_matmul_cuda.py
test_meta.py Added meta function for _upsample_bicubic2d_aa (#117347) 2024-01-16 23:33:55 +00:00
test_metal.py
test_mkl_verbose.py
test_mkldnn.py [BE]: Enable RUFF PERF402 and apply fixes (#115505) 2023-12-20 18:01:24 +00:00
test_mkldnn_fusion.py [codemod] markDynamoStrictTest batch 22 (#117729) 2024-01-18 16:59:26 +00:00
test_mkldnn_verbose.py
test_mobile_optimizer.py Revert "markDynamoStrictTest on more tests (#115879)" 2023-12-15 20:00:09 +00:00
test_model_dump.py
test_model_exports_to_core_aten.py
test_module_init.py Revert "markDynamoStrictTest on more tests (#115879)" 2023-12-15 20:00:09 +00:00
test_modules.py Revert "markDynamoStrictTest on more tests (#115879)" 2023-12-15 20:00:09 +00:00
test_monitor.py Change dynamo_test_failures.py to silently run skipped tests (#117401) 2024-01-17 02:48:19 +00:00
test_mps.py add Half support for interpolate operators on CPU (#105648) 2024-01-18 09:07:16 +00:00
test_multiprocessing.py Revert "markDynamoStrictTest on more tests (#115879)" 2023-12-15 20:00:09 +00:00
test_multiprocessing_spawn.py Revert "markDynamoStrictTest on more tests (#115879)" 2023-12-15 20:00:09 +00:00
test_namedtensor.py
test_namedtuple_return_api.py Revert "[CUDNN][SDPA] Experimental cuDNN Flash Attention v2 Inference (#115663)" 2024-01-18 23:40:30 +00:00
test_native_functions.py
test_native_mha.py Re-add initial Flash Attention support on ROCM (#115981) 2024-01-04 22:21:31 +00:00
test_nestedtensor.py [Nested Tensor]Support SDPA math fallback for jagged layout nested tensor (#116445) 2024-01-12 17:30:40 +00:00
test_nn.py Remove skips for passing tests (#118000) 2024-01-23 16:11:38 +00:00
test_nnapi.py
test_numba_integration.py
test_numpy_interop.py Improve uint{16,32,64} dlpack/numpy compatibility (#116808) 2024-01-11 17:01:54 +00:00
test_openmp.py
test_ops.py Initial implementation of FakeTensor caching (#113873) 2024-01-17 20:38:54 +00:00
test_ops_fwd_gradients.py unMarkDynamoStrictTest on OpInfo-based tests (#115856) 2023-12-15 01:22:31 +00:00
test_ops_gradients.py unMarkDynamoStrictTest on OpInfo-based tests (#115856) 2023-12-15 01:22:31 +00:00
test_ops_jit.py unMarkDynamoStrictTest on OpInfo-based tests (#115856) 2023-12-15 01:22:31 +00:00
test_optim.py Migrate test_complex_optimizer to OptimizerInfo (#118160) 2024-01-24 21:22:47 +00:00
test_out_dtype_op.py
test_overrides.py [BE]: Update flake8 to v6.1.0 and fix lints (#116591) 2024-01-03 06:04:44 +00:00
test_package.py
test_per_overload_api.py
test_prims.py Revert "markDynamoStrictTest some more (#115885)" 2023-12-15 19:51:24 +00:00
test_proxy_tensor.py Always accept 0-d scalar tensors as int, even if __index__ fails (#117451) 2024-01-14 14:39:46 +00:00
test_pruning_op.py
test_public_bindings.py [ONNX][dynamo_export] Decomposition skips using custom operator (#117314) 2024-01-18 22:19:28 +00:00
test_python_dispatch.py [BE]: Update flake8 to v6.1.0 and fix lints (#116591) 2024-01-03 06:04:44 +00:00
test_pytree.py [pytree] add access api (#117771) 2024-01-20 04:03:26 +00:00
test_quantization.py
test_reductions.py Make and/or on uint8 tensors properly return 0x00 or 0x01 (#117827) 2024-01-22 17:30:22 +00:00
test_scatter_gather_ops.py
test_schema_check.py Revert "markDynamoStrictTest some more (#115885)" 2023-12-15 19:51:24 +00:00
test_segment_reductions.py
test_serialization.py [CI] Install dill in ci (#116214) 2024-01-24 23:42:35 +00:00
test_set_default_mobile_cpu_allocator.py
test_shape_ops.py
test_show_pickle.py Revert "markDynamoStrictTest some more (#115885)" 2023-12-15 19:51:24 +00:00
test_sort_and_select.py Removed an internal assertion for the optional stable value and inste… (#117414) 2024-01-17 02:25:21 +00:00
test_sparse.py Add batched sparse CSR/CSC/BSR/BSC to sparse COO conversion support (#116206) 2024-01-07 19:42:02 +00:00
test_sparse_csr.py [ROCm] skip test_sparse_csr test_triton_bsr_softmax_cuda (#118006) 2024-01-23 00:09:42 +00:00
test_sparse_semi_structured.py Add out_dtype support for sparse semi-structured CUTLASS back-end (#116519) 2024-01-03 16:23:17 +00:00
test_spectral_ops.py FFT: Handle noop fftn calls gracefully (#117368) 2024-01-12 20:16:50 +00:00
test_stateless.py
test_static_runtime.py
test_subclass.py
test_sympy_utils.py
test_tensor_creation_ops.py Enable some uint{16,32,64} tests that are working (#116809) 2024-01-15 02:25:21 +00:00
test_tensorboard.py
test_tensorexpr.py
test_tensorexpr_pybind.py
test_testing.py Back out "[Kineto] Initialize libkineto profilers during torch init process during pybind set-up (#112623)" (#116201) 2023-12-21 16:32:19 +00:00
test_throughput_benchmark.py
test_torch.py Fix swap_tensors to swap PyObjects associated with TensorImpl (#116955) 2024-01-24 01:40:18 +00:00
test_transformers.py Remove sdp_kernel and replace with sdpa_kernel in attention namespace (#114689) 2024-01-24 22:28:04 +00:00
test_type_hints.py
test_type_info.py
test_type_promotion.py Improve uint{16,32,64} dlpack/numpy compatibility (#116808) 2024-01-11 17:01:54 +00:00
test_typing.py
test_unary_ufuncs.py Remove incorrect usages of skipIfTorchDynamo (#117114) 2024-01-10 22:25:31 +00:00
test_utils.py Some tiny modification about torch.set/get_default_device (#116014) 2023-12-19 05:08:06 +00:00
test_view_ops.py
test_vulkan.py
test_weak.py
test_xnnpack_integration.py