pytorch/torch/utils
2025-06-10 04:42:06 +00:00
..
_strobelight
_sympy
backcompat
benchmark [BE]: Backport runtime_checkable perf improvements/behavior from 3.12 (#155130) 2025-06-06 13:28:05 +00:00
bottleneck
data remove allow-untyped-defs from torch/utils/data/datapipes/iter/filelister.py (#154624) 2025-05-30 08:38:05 +00:00
hipify [BE] Delete pre-CUDA-10.1 code from SparseCUDABlas (#155079) 2025-06-04 03:29:24 +00:00
jit
model_dump
serialization
tensorboard
viz [Visualizer] Start at index with most events (#154571) 2025-05-29 20:49:33 +00:00
__init__.py
_appending_byte_serializer.py
_backport_slots.py
_config_module.py Revert "inductor codecache: include private inductor configs in cache key (#153672)" 2025-05-30 15:54:14 +00:00
_config_typing.pyi Revert "inductor codecache: include private inductor configs in cache key (#153672)" 2025-05-30 15:54:14 +00:00
_content_store.py
_contextlib.py
_cpp_embed_headers.py
_cpp_extension_versioner.py
_cxx_pytree.py
_device.py
_dtype_abbrs.py
_exposed_in.py
_filelock.py
_foreach_utils.py
_freeze.py
_functools.py
_get_clean_triton.py [BE]Enhance _get_clean_triton.py to auto-generate launch_params if missing (#154666) 2025-05-31 19:27:56 +00:00
_import_utils.py
_mode_utils.py
_ordered_set.py
_python_dispatch.py
_pytree.py Preserve Enum types during torch.export serialization and deserialization (#154821) 2025-06-08 17:30:31 +00:00
_stats.py
_thunk.py
_traceback.py
_triton.py Revert "Update auto-tuning support for _scaled_grouped_mm (#150944)" 2025-06-09 23:12:56 +00:00
_typing_utils.py
_zip.py
backend_registration.py
bundled_inputs.py
checkpoint.py [FlexAttention] Allow dispatch to SAC for flex (#150080) 2025-06-05 04:34:27 +00:00
collect_env.py Revert "Add Intel GPU info collection to the collect env script (#137846)" 2025-06-09 03:13:27 +00:00
cpp_backtrace.py
cpp_extension.py [ROCm] cpp_extension allow user to override default flags (#152432) 2025-05-15 21:06:18 +00:00
deterministic.py
dlpack.py
file_baton.py
flop_counter.py Inductor logging + analysis of torch.profile (#149697) 2025-06-09 21:43:21 +00:00
hooks.py Add warning for module full backward hook when no input requires gradient (#155339) 2025-06-10 04:42:06 +00:00
mkldnn.py
mobile_optimizer.py
model_zoo.py
module_tracker.py
show_pickle.py
throughput_benchmark.py
weak.py pymft lint torch/utils/weak.py (#154484) 2025-05-28 17:06:58 +00:00