pytorch/test
BowenBao dc0071dfa5 [ONNX] Special post process for onnx::Cast and onnx::ConstantOfShape shape type inference (#55962) (#57597)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/57597

* Special post process for onnx::Cast and onnx::ConstantOfShape
* Update `test_pytorch_onnx_shape_inference.py` to be unit test over shape inference patterns.

Test Plan: Imported from OSS

Reviewed By: malfet

Differential Revision: D28393529

Pulled By: SplitInfinity

fbshipit-source-id: fc26032ddb842d4e299447da39564b28049752ed

Co-authored-by: BowenBao <bowbao@microsoft.com>
2021-05-13 13:42:44 -07:00
..
backward_compatibility Striding for lists Part 2 (#49352) 2021-05-13 00:16:02 -07:00
benchmark_utils Add lint for unqualified noqa (#56272) 2021-04-19 13:16:18 -07:00
bottleneck_test
cpp [Pytorch Edge] Runtime ops compatibility api (#57570) 2021-05-13 10:20:41 -07:00
cpp_api_parity Remove tmp folder when run unit test (#57800) 2021-05-09 20:07:14 -07:00
cpp_extensions Fix UB in library.h (#57962) 2021-05-11 16:04:02 -07:00
custom_backend
custom_operator AutoDispatchBelowAutograd takes no arguments. (#56424) 2021-04-21 14:44:12 -07:00
distributed Fix multi gpu test break on Windows (#58213) 2021-05-12 21:39:08 -07:00
distributions Deprecate torch.cholesky (#57725) 2021-05-10 22:44:25 -07:00
error_messages
expect Enable backward/forward compatibility for TS runtime (#57498) 2021-05-07 15:41:45 -07:00
fx Maintain submodule references during subgraph rewriting (#55463) 2021-04-30 16:46:44 -07:00
jit Add support for real and imag tensor attributes (#54692) 2021-05-13 10:44:27 -07:00
jit_hooks
mobile [clang-tidy] Exclude cppcoreguidelines-avoid-magic-numbers (#57841) 2021-05-07 20:02:33 -07:00
onnx [ONNX] Special post process for onnx::Cast and onnx::ConstantOfShape shape type inference (#55962) (#57597) 2021-05-13 13:42:44 -07:00
optim
package [package] Add an intern keyword (#57341) 2021-05-12 16:22:43 -07:00
quantization Fixing quantize_per_tensor on cuda (#57703) 2021-05-07 12:26:19 -07:00
scripts
test_img
typing Deprecate QR (#57745) 2021-05-10 22:56:37 -07:00
HowToWriteTestsUsingFileCheck.md
linear.py
run_test.py Add Python-3.9 CI testing (#50992) 2021-05-10 10:51:39 -07:00
simulate_nccl_errors.py
test_ao_sparse.py [sparsity] Moving the sparsity python files to OSS (#56617) 2021-04-22 14:07:31 -07:00
test_autograd.py Revert "Revert D28387764: Codegen inplace forward AD formula from out of place one if needed" (#58231) 2021-05-13 13:18:16 -07:00
test_binary_ufuncs.py Port heaviside to structured kernel (#57933) 2021-05-13 10:48:11 -07:00
test_bundled_images.py
test_bundled_inputs.py Add lint for unqualified type: ignore (#56290) 2021-04-21 08:07:23 -07:00
test_complex.py
test_cpp_api_parity.py
test_cpp_extensions_aot.py
test_cpp_extensions_jit.py
test_cuda.py [CUDA graphs] Sync after replay (#57556) 2021-05-11 09:38:47 -07:00
test_cuda_primary_ctx.py
test_dataloader.py [DataLoader] Add generate_state for NumPy seeding (#56797) 2021-04-27 08:14:02 -07:00
test_datapipe.py [DataLoader] Add context manager for runtime type validation (#55936) 2021-05-12 11:59:16 -07:00
test_determination.py
test_dispatch.py Add lint for unqualified noqa (#56272) 2021-04-19 13:16:18 -07:00
test_expecttest.py
test_foreach.py
test_function_schema.py Add stricter check for function schemas with varargs (#56509) 2021-04-20 20:04:38 -07:00
test_functional_autograd_benchmark.py
test_futures.py Add lint for unqualified type: ignore (#56290) 2021-04-21 08:07:23 -07:00
test_fx.py Revert D28351931: [pytorch][PR] Fix some tensor operators to return NotImplemented for invalid inputs 2021-05-12 13:58:03 -07:00
test_fx_experimental.py Revert D28351931: [pytorch][PR] Fix some tensor operators to return NotImplemented for invalid inputs 2021-05-12 13:58:03 -07:00
test_import_time.py
test_indexing.py Migrate thrust->cub for index put (#55693) 2021-04-27 12:27:09 -07:00
test_jit.py Striding for lists Part 2 (#49352) 2021-05-13 00:16:02 -07:00
test_jit_cuda_fuser.py Fixing error in jit cuda on ROCm: non-constant-expression cannot be n… (#55243) 2021-04-21 16:35:27 -07:00
test_jit_disabled.py
test_jit_fuser.py
test_jit_fuser_legacy.py
test_jit_fuser_te.py [TensorExpr] Remove disabled tests that we do not plan to re-enable. (#58207) 2021-05-13 09:19:20 -07:00
test_jit_legacy.py
test_jit_profiling.py
test_jit_simple.py
test_jit_string.py
test_kernel_launch_checks.py T90561249: Enforce kernel launch checks (#58178) 2021-05-13 10:41:20 -07:00
test_license.py
test_linalg.py Fix NumPy compatibility issue for torch.linalg.cond (#58041) 2021-05-13 09:42:18 -07:00
test_logging.py
test_metal.py
test_mkldnn.py [BE] Improve has_bf16_support (#57408) 2021-05-03 09:11:04 -07:00
test_mobile_optimizer.py [Pytorch Edge] Remove methods_to_optimize arg (#57045) 2021-04-27 14:54:13 -07:00
test_model_dump.py model_dump tool for model inspection (#56868) 2021-04-28 07:33:10 -07:00
test_module_init.py Support factory kwargs in torch.nn modules (#54508) 2021-04-22 16:16:53 -07:00
test_multiprocessing.py Fix mp serialization for integer nn.Parameter on CUDA (#56529) 2021-04-22 16:21:04 -07:00
test_multiprocessing_spawn.py
test_namedtensor.py
test_namedtuple_return_api.py Add torch.linalg.inv_ex without checking for errors by default (#58039) 2021-05-13 09:42:15 -07:00
test_native_functions.py
test_nn.py Deprecate torch.solve (#57741) 2021-05-13 09:54:21 -07:00
test_nnapi.py NNAPI: flex size support for upsample_nearest2d op (#57563) 2021-05-05 13:54:43 -07:00
test_numba_integration.py
test_numpy_interop.py Add lint for unqualified type: ignore (#56290) 2021-04-21 08:07:23 -07:00
test_op_aliases.py [OpInfo] Add opinfo for transpose and its aliases (#56122) 2021-04-25 21:58:16 -07:00
test_openmp.py
test_ops.py Revert D28387767: Add forward AD test for op info 2021-05-12 20:41:25 -07:00
test_optim.py
test_overrides.py Revert "Revert D28387764: Codegen inplace forward AD formula from out of place one if needed" (#58231) 2021-05-13 13:18:16 -07:00
test_package.py
test_profiler.py Update Kineto submodule (#58179) 2021-05-13 04:03:04 -07:00
test_pruning_op.py
test_public_bindings.py
test_pytree.py [FX] Adds PyTree support to FX through concrete_args (#55888) 2021-05-07 04:48:35 -07:00
test_quantization.py Add lint for unqualified noqa (#56272) 2021-04-19 13:16:18 -07:00
test_reductions.py std/var: Return real results for complex input (#58066) 2021-05-12 03:26:55 -07:00
test_segment_reductions.py [torch][segment_reduce] Add support for initial value (#56923) 2021-04-30 18:01:31 -07:00
test_serialization.py
test_set_default_mobile_cpu_allocator.py
test_shape_ops.py Fix tensor device in test_kthvalue_overlap (#56869) 2021-05-01 12:01:09 -07:00
test_show_pickle.py
test_sort_and_select.py fix topk with k=0 on cuda (#58086) 2021-05-12 13:38:10 -07:00
test_sparse.py New: sparse complex: add linear algebra, addmm (#57129) 2021-05-07 05:37:48 -07:00
test_sparse_csr.py Modernize test-suite in sparse tensor CSR (#56392) 2021-04-27 15:22:17 -07:00
test_spectral_ops.py Remove distutils (#57040) 2021-04-29 12:10:11 -07:00
test_static_runtime.py
test_tensor_creation_ops.py Fix legacy tensor constructor/new matching incorrect signature with d… (#58108) 2021-05-13 08:11:08 -07:00
test_tensorboard.py Add lint for unqualified noqa (#56272) 2021-04-19 13:16:18 -07:00
test_tensorexpr.py [TensorExpr] Re-enable skipped tests, they seem to be working now. (#58206) 2021-05-13 09:18:09 -07:00
test_tensorexpr_pybind.py Fix lint in test_tensorexpr_pybind (#57869) 2021-05-07 15:58:21 -07:00
test_testing.py enable support for nested containers in torch.testing.assert(equal|close) (#57270) 2021-05-12 15:37:42 -07:00
test_throughput_benchmark.py
test_torch.py OpInfo: squeeze (#58080) 2021-05-12 21:29:31 -07:00
test_type_hints.py Add lint for unqualified type: ignore (#56290) 2021-04-21 08:07:23 -07:00
test_type_info.py
test_type_promotion.py New: Initial support for sparse complex tensors constructors for CPU/CUDA (#57125) 2021-05-07 05:36:41 -07:00
test_typing.py
test_unary_ufuncs.py Enable geometric ops, exp2, expm1, rsqrt & erfc for BFloat16 on CUDA (#57913) 2021-05-11 09:43:05 -07:00
test_utils.py Add lint for unqualified type: ignore (#56290) 2021-04-21 08:07:23 -07:00
test_view_ops.py Adding hsplit,vsplit and dsplit methods (#53536) 2021-04-26 09:39:09 -07:00
test_vmap.py
test_vulkan.py
test_xnnpack_integration.py