Stephen Jia c2e7a0d689 [core IR] Add decomps for aten.sum and aten.squeeze variants (#110645) ... Summary: ## Context Both `aten.sum` and `aten.squeeze` have a "most generic" variant in the form of `aten.sum.dim_IntList` and `aten.squeeze.dims` respectively. Add decompositions for other non generic variants of these operators to express them using the most generic variant. Note that to register these decomps, the reference implementation under `_refs` had to be removed as registered decompositions. cc: @lezcano @peterbell10 Test Plan: Github CI + Meta Internal CI Differential Revision: D49965952 Pull Request resolved: https://github.com/pytorch/pytorch/pull/110645 Approved by: https://github.com/peterbell10, https://github.com/digantdesai, https://github.com/manuelcandales		2023-10-07 04:21:51 +00:00
..
__init__.py
HasDecompTest.test_aten_core_operators.expect	[core IR] Add decomps for aten.sum and aten.squeeze variants (#110645)	2023-10-07 04:21:51 +00:00
HasDecompTest.test_has_decomposition.expect	Fix python decomps for OpOverloadPackets and add tests (#107707)	2023-09-25 20:53:30 +00:00
TestAutograd.test_function-x_grad_desc.expect
TestAutograd.test_function-y_grad_desc.expect
TestFXAPIBackwardCompatibility.test_class_member_back_compat-fx_backcompat_class_members.expect	Refactor repro/minifier into CLI; add analyze (#100226)	2023-05-01 11:12:38 +00:00
TestFXAPIBackwardCompatibility.test_function_back_compat-fx_backcompat_function_signatures.expect	Refactor repro/minifier into CLI; add analyze (#100226)	2023-05-01 11:12:38 +00:00
TestJit.test_cu_escaped_number.expect
TestJit.test_import_method.expect
TestJit.test_non_ascii_string.expect
TestJit.test_pretty_printer-empty_float_list_test.expect
TestJit.test_pretty_printer-empty_int_list_test.expect
TestJit.test_pretty_printer-if_one.expect
TestJit.test_pretty_printer-if_test.expect
TestJit.test_pretty_printer-loop_use_test.expect
TestJit.test_pretty_printer-print_weird_test.expect
TestJit.test_pretty_printer-python_op_name_test.expect
TestJit.test_pretty_printer-while_if_test.expect
TestJit.test_pretty_printer-while_test.expect
TestPytorchExportModes.test_aten_fallback.expect
TestPytorchExportModes.test_onnx_aten.expect
TestScript.test_annot_ast_mypy_fn.expect
TestScript.test_annot_ast_mypy_method.expect
TestScript.test_annot_ast_py3_fn.expect
TestScript.test_annot_ast_py3_method.expect
TestScript.test_annot_string_mypy_fn.expect
TestScript.test_annot_string_mypy_method.expect
TestScript.test_annot_string_py3_fn.expect
TestScript.test_annot_string_py3_method.expect
TestScript.test_annotated_script_fn.expect
TestScript.test_annotated_script_method.expect
TestScript.test_format-stdout.expect
TestScript.test_listconstruct_erasure.expect
TestScript.test_parser_type_annotations.expect
TestScript.test_parser_type_annotations_comment.expect
TestScript.test_print-stdout.expect	Remove set_default_dtype calls from jit and ops tests (#105072)	2023-07-15 03:18:33 +00:00
TestScript.test_python_frontend.expect
TestScript.test_python_frontend_py2.expect
TestScript.test_python_frontend_py3.expect
TestScript.test_string_print-stdout.expect
TestScript.test_torch_dot_tensor_annotation.expect
TestSparseCompressedCPU.test_print_SparseBSC_cpu.expect
TestSparseCompressedCPU.test_print_SparseBSR_cpu.expect
TestSparseCompressedCPU.test_print_SparseCSC_cpu.expect
TestSparseCompressedCPU.test_print_SparseCSR_cpu.expect
TestSparseCompressedCUDA.test_print_SparseBSC_cuda.expect
TestSparseCompressedCUDA.test_print_SparseBSR_cuda.expect
TestSparseCompressedCUDA.test_print_SparseCSC_cuda.expect
TestSparseCompressedCUDA.test_print_SparseCSR_cuda.expect
TestSparseCPU.test_print_coalesced_cpu_float64.expect
TestSparseCPU.test_print_uncoalesced_cpu_float64.expect
TestSparseCUDA.test_print_coalesced_cuda_float64.expect
TestSparseCUDA.test_print_uncoalesced_cuda_float64.expect
TestTensorBoard.test_audio.expect
TestTensorBoard.test_caffe2_simple_cnnmodel.expect
TestTensorBoard.test_caffe2_simple_model.expect
TestTensorBoard.test_histogram_auto.expect
TestTensorBoard.test_histogram_doane.expect
TestTensorBoard.test_histogram_fd.expect
TestTensorBoard.test_hparams_bool.expect
TestTensorBoard.test_hparams_number.expect
TestTensorBoard.test_hparams_string.expect
TestTensorBoard.test_image_with_3_channel_batched.expect
TestTensorBoard.test_image_with_boxes.expect
TestTensorBoard.test_image_with_one_channel.expect
TestTensorBoard.test_image_with_one_channel_batched.expect
TestTensorBoard.test_image_without_channel.expect
TestTensorBoard.test_mesh.expect
TestTensorBoard.test_nested_nn_squential.expect
TestTensorBoard.test_pr_curve.expect
TestTensorBoard.test_pr_curve_raw.expect
TestTensorBoard.test_pytorch_graph.expect
TestTensorBoard.test_scalar_new_style.expect
TestTensorBoard.test_text.expect
TestTensorBoard.test_video.expect
TestTorch.test_is_nonzero-empty.expect
TestTorch.test_is_nonzero-multiple.expect
TestTorch.test_print-non_contiguous.expect