pytorch/torch/utils
2025-06-11 19:04:10 +00:00
..
_strobelight
_sympy [inductor] Fix ModularIndexing assumptions (#152993) 2025-05-08 18:26:45 +00:00
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 inductor codecache: include private inductor configs in cache key (#153672) 2025-06-11 01:33:24 +00:00
_config_typing.pyi inductor codecache: include private inductor configs in cache key (#153672) 2025-06-11 01:33:24 +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
_helion.py Add helion x pt2 test (#155513) 2025-06-11 07:08:06 +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 Add types to torch/utils/_triton.py (#155612) 2025-06-11 19:04:10 +00:00
_typing_utils.py
_zip.py
backend_registration.py
bundled_inputs.py
checkpoint.py [BE] Mention debug=True in AC error messages (#155593) 2025-06-11 00:32:41 +00:00
collect_env.py Revert "Add Intel GPU info collection to the collect env script (#137846)" 2025-06-11 15:18:47 +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 Revert "Inductor logging + analysis of torch.profile (#149697)" 2025-06-10 15:38:40 +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