Tugsbayasgalan Manlaibaatar 4f120dc2a6 Clean up mode handling in python dispatcher (#121083) ... Things that were bad before this PR: 1. Temporarily unsetting functional tensor mode and proxy mode both had duplicate implementation 2. There are variants of mode handling private utils that has duplicate implementation. (different APIs calling repeated implementation, so i refactored) 3. _push_mode API used to take dispatch key argument which is not necessary. 4. There are unused APIs. Pull Request resolved: https://github.com/pytorch/pytorch/pull/121083 Approved by: https://github.com/zou3519		2024-03-08 00:30:34 +00:00
..
_sympy	Skip less replacements (#119570)	2024-02-20 04:10:33 +00:00
backcompat
benchmark	Enable possibly-undefined error code (#118533)	2024-01-30 21:07:01 +00:00
bottleneck
data	Preserve metadata for MutableMapping and MutableSequence in pin_memory and collate_fn (#120553)	2024-03-01 20:43:42 +00:00
hipify	[ROCm] Enable float16/complex32 fft tests on ROCm (#117296)	2024-02-13 22:35:32 +00:00
jit
model_dump
tensorboard	[tensorboard] Handle bfloat16 type in add_histogram (#120087)	2024-03-05 00:27:21 +00:00
viz	[Memory Snapshot] Add Total memory used after allocation in Trace View (#120339)	2024-02-23 16:17:14 +00:00
__init__.py
_config_module.py	Enable local_partial_types (#118467)	2024-01-28 13:38:22 +00:00
_config_typing.pyi
_content_store.py
_contextlib.py
_cpp_extension_versioner.py
_cuda_trace.py
_cxx_pytree.py	[pytree] support X | Y union type in tree_map_only (#120389)	2024-02-22 18:17:13 +00:00
_device.py	Enable local_partial_types (#118467)	2024-01-28 13:38:22 +00:00
_foreach_utils.py	[DTensor] Supported foreach=True for clip_grad_norm_ (#120910)	2024-03-02 00:28:09 +00:00
_freeze.py
_import_utils.py	[CI] Install dill in ci (#116214)	2024-01-24 23:42:35 +00:00
_mode_utils.py
_python_dispatch.py	Clean up mode handling in python dispatcher (#121083)	2024-03-08 00:30:34 +00:00
_pytree.py	[pytree][reland] Require pytree serialized_type_name (#120636)	2024-02-27 06:53:33 +00:00
_stats.py
_traceback.py
_triton.py	Add an autotune cache for inductor generated kernels (#120963)	2024-03-04 16:58:37 +00:00
_typing_utils.py
_zip.py
backend_registration.py
bundled_inputs.py
checkpoint.py	Enable possibly-undefined error code (#118533)	2024-01-30 21:07:01 +00:00
collect_env.py
cpp_backtrace.py
cpp_extension.py	Fix venv compatibility issue by updating python_lib_path (#121103)	2024-03-06 17:00:46 +00:00
deterministic.py
dlpack.py
file_baton.py
flop_counter.py	fixed flop counter formula for conv transposed backwards pass (#119874)	2024-02-16 02:43:49 +00:00
hooks.py
mkldnn.py
mobile_optimizer.py
model_zoo.py
show_pickle.py
throughput_benchmark.py
weak.py