Sergii Dymchenko de7219e8a7 Use generators with all/any in torch/optim (#78142) ... Generator comprehensions with any/all are less verbose and potentially help to save memory/CPU : https://eklitzke.org/generator-comprehensions-and-using-any-and-all-in-python To make JIT work with this change, I added code to convert GeneratorExp to ListComp. So the whole PR is basically NoOp for JIT, but potentially memory and speed improvement for eager mode. Also I removed a test from test/jit/test_parametrization.py. The test was bad and had a TODO to actually implement and just tested that UnsupportedNodeError is thrown, and with GeneratorExp support a different error would be thrown. Pull Request resolved: https://github.com/pytorch/pytorch/pull/78142 Approved by: https://github.com/malfet, https://github.com/albanD		2022-06-24 17:23:45 +00:00
..
_imported_class_test
fixtures	Revert "stft: remove non-center overload and python functional wrapper"	2022-05-09 19:59:46 +00:00
fixtures_srcs	Revert "stft: remove non-center overload and python functional wrapper"	2022-05-09 19:59:46 +00:00
__init__.py
myexception.py
test_alias_analysis.py
test_async.py
test_aten_pow.py
test_attr.py
test_autodiff.py	Retry - [JIT] Propagate profiled information to DifferentiableGraph outputs	2022-06-21 22:57:17 +00:00
test_autodiff_subgraph_slicing.py	Retry - [JIT] Propagate profiled information to DifferentiableGraph outputs	2022-06-21 22:57:17 +00:00
test_backend_nnapi.py
test_backends.py
test_batch_mm.py
test_builtins.py
test_class_type.py	Register PrimTorch refs as decompositions.	2022-05-06 20:11:45 +00:00
test_complex.py
test_complexity.py
test_convert_activation.py
test_cuda.py
test_custom_operators.py
test_data_parallel.py
test_dataclasses.py	Reland "[pytorch][PR] Support dataclasses in TorchScript" take 2 (#74353) (#74353) (#76771)	2022-06-07 21:44:55 +00:00
test_dce.py
test_device_analysis.py	[EASY] [JIT]Cleaning up the device analysis temp test code	2022-04-25 15:50:52 +00:00
test_dtype_analysis.py
test_enum.py
test_exception.py
test_export_modes.py
test_freezing.py	[JIT] enable autocasting + freezing test	2022-06-02 22:38:56 +00:00
test_functional_blocks.py
test_fuser_common.py
test_graph_rewrite_passes.py
test_hash.py
test_hooks.py
test_hooks_modules.py
test_ignorable_args.py
test_ignore_context_manager.py
test_isinstance.py
test_ivalue.py	Add missing QSCheme IValue conversion logic (#78862)	2022-06-07 08:34:17 +00:00
test_jit_utils.py	Disable TracerWarnings on NNC opinfo tests	2022-06-03 18:11:12 +00:00
test_legacy_upgraders.py
test_list_dict.py
test_logging.py
test_misc.py	Reland "[pytorch][PR] Support dataclasses in TorchScript" take 2 (#74353) (#74353) (#76771)	2022-06-07 21:44:55 +00:00
test_models.py	add channels last support for thnn_conv2d (non-dilated)	2022-05-11 00:09:45 +00:00
test_module_apis.py
test_module_containers.py	[JIT] Support for ParameterDict getattr	2022-05-13 01:03:25 +00:00
test_module_interface.py
test_modules.py
test_op_decompositions.py	Reland reland	2022-04-28 20:41:55 +00:00
test_optimize_for_mobile_preserve_debug_info.py
test_parametrization.py	Use generators with all/any in torch/optim (#78142)	2022-06-24 17:23:45 +00:00
test_pdt.py
test_peephole.py
test_profiler.py
test_python_bindings.py
test_python_builtins.py
test_python_ir.py
test_recursive_script.py
test_remove_mutation.py	functionalization: add native fill() op	2022-04-25 21:34:16 +00:00
test_save_load.py	[PyTorchEdge] Extend Flatbuffer to get mobile_info for NMLML workflows	2022-06-01 20:09:09 +00:00
test_save_load_for_op_version.py	Revert "stft: remove non-center overload and python functional wrapper"	2022-05-09 19:59:46 +00:00
test_schema_check.py	[JIT] Add checks for outputs aliasing unexpectedly, implement logging and flesh out test suite (#79904)	2022-06-23 20:42:07 +00:00
test_script_profile.py
test_scriptmod_ann.py
test_slice.py
test_sparse.py
test_string_formatting.py
test_symbolic_shape_analysis.py	fix index_select when source tensor is empty	2022-05-26 03:10:47 +00:00
test_tensor_creation_ops.py
test_tensor_methods.py
test_torchbind.py
test_tracer.py	Move functorch decompositions to PyTorch	2022-04-30 16:47:53 +00:00
test_type_sharing.py
test_types.py	[JIT] Improve error message for InferredType	2022-06-08 23:02:58 +00:00
test_typing.py	Register PrimTorch refs as decompositions.	2022-05-06 20:11:45 +00:00
test_union.py
test_unsupported_ops.py
test_upgraders.py	Updates floor_divide to perform floor division (#78411)	2022-05-29 21:28:45 +00:00
test_warn.py
test_with.py