pytorch/torch/utils
Henry Hu f013cfee38 [TreeSpec] Support enum in defaultdict (#144235)
Summary: Followup from D66269157, add support for enum in defaultdict.

Test Plan: Added unit test

Differential Revision: D67832100

Pull Request resolved: https://github.com/pytorch/pytorch/pull/144235
Approved by: https://github.com/henrylhtsang, https://github.com/houseroad
2025-01-07 00:10:46 +00:00
..
_strobelight Propagate callable parameter types using ParamSpec (#142306) (#143797) 2024-12-29 23:03:14 +00:00
_sympy Fix PythonMod printing (#144078) 2025-01-06 22:52:34 +00:00
backcompat
benchmark remove allow-untyped-defs for torch/utils/benchmark/examples/simple_timeit.py (#143368) 2024-12-17 17:19:11 +00:00
bottleneck
data Revert "Use absolute path path.resolve() -> path.absolute() (#129409)" 2025-01-04 14:17:20 +00:00
hipify [reland][attempt2][AMD] Turn on TF32 for aten::mm (#144145) 2025-01-06 00:37:01 +00:00
jit
model_dump [BE] Remove Model Dump utility (#141540) 2024-11-27 22:52:55 +00:00
serialization Add config.save.use_pinned_memory_for_d2h to serialization config (#143342) 2024-12-20 21:01:18 +00:00
tensorboard remove allow-untyped-defs from utils/tensorboard/_convert_np.py (#143918) 2024-12-27 22:19:33 +00:00
viz
__init__.py
_backport_slots.py
_config_module.py Add config alias (#142088) 2024-12-16 18:51:17 +00:00
_config_typing.pyi
_content_store.py
_contextlib.py
_cpp_extension_versioner.py
_cxx_pytree.py [BE][CI] bump ruff to 0.8.4 (#143753) 2024-12-24 12:24:10 +00:00
_device.py [BE]: Apply PERF401 autofixes from ruff (#140980) 2024-11-20 17:52:07 +00:00
_exposed_in.py Type exposed_in decorator (#141894) 2024-12-03 16:28:17 +00:00
_filelock.py filelock: Make waitcounter variant to use (#139816) 2024-12-12 01:18:34 +00:00
_foreach_utils.py
_freeze.py [BE][CI] bump ruff to 0.8.4 (#143753) 2024-12-24 12:24:10 +00:00
_functools.py dynamo tracing perf: cache on import_source: 52.9 -> 52.58 (#143058) 2024-12-13 18:20:48 +00:00
_get_clean_triton.py
_import_utils.py remove allow-untyped-defs from utils/_import_utils.py (#144089) 2025-01-03 01:21:13 +00:00
_mode_utils.py
_ordered_set.py
_python_dispatch.py
_pytree.py [TreeSpec] Support enum in defaultdict (#144235) 2025-01-07 00:10:46 +00:00
_stats.py Propagate callable parameter types using ParamSpec (#142306) (#143797) 2024-12-29 23:03:14 +00:00
_thunk.py
_traceback.py
_triton.py Adding lowering to persistent-tma device kernel for _scaled_mm (#142045) 2024-12-09 01:48:40 +00:00
_typing_utils.py
_zip.py
backend_registration.py
bundled_inputs.py
checkpoint.py [BE] typing for decorators (#144161) 2025-01-04 16:40:09 +00:00
collect_env.py Disable PIP version check in collect_env (#142308) 2024-12-10 19:16:36 +00:00
cpp_backtrace.py
cpp_extension.py Revert "Use absolute path path.resolve() -> path.absolute() (#129409)" 2025-01-04 14:17:20 +00:00
deterministic.py
dlpack.py
file_baton.py
flop_counter.py [BE] typing for decorators (#144161) 2025-01-04 16:40:09 +00:00
hooks.py [BE]: Apply PERF401 autofixes from ruff (#140980) 2024-11-20 17:52:07 +00:00
mkldnn.py
mobile_optimizer.py [BE]: Apply PERF401 autofixes from ruff (#140980) 2024-11-20 17:52:07 +00:00
model_zoo.py
module_tracker.py Fix memory leak in ModuleTracker (#141960) 2024-12-03 18:36:15 +00:00
show_pickle.py
throughput_benchmark.py
weak.py