pytorch/torch/utils
2025-05-17 06:21:41 +00:00
..
_strobelight
_sympy [inductor] Fix ModularIndexing assumptions (#152993) 2025-05-08 18:26:45 +00:00
backcompat
benchmark Clean up conda usage in benchmark scripts (#152552) 2025-04-30 21:27:29 +00:00
bottleneck
data Refactor torch/utils/data/datapipes/gen_pyi.py with torchgen (#150626) 2025-05-17 06:21:41 +00:00
hipify Add cudaLaunchKernel to cuda_to_hip_mappings (#153690) 2025-05-16 23:37:11 +00:00
jit
model_dump
serialization
tensorboard Fix broken URLs (#152237) 2025-04-27 09:56:42 +00:00
viz [MemoryZ] Show the current and max entries rendered (#153446) 2025-05-14 03:16:12 +00:00
__init__.py
_appending_byte_serializer.py Check integrity of bytes in AppendingByteSerializer (#152139) 2025-04-26 18:10:58 +00:00
_backport_slots.py
_config_module.py [dynamo] context manager/decorator for dynamo config patching during tracing (#150586) 2025-04-23 09:12:13 +00:00
_config_typing.pyi
_content_store.py
_contextlib.py
_cpp_embed_headers.py
_cpp_extension_versioner.py
_cxx_pytree.py [BE] detect CXX pytree requirement with TorchVersion (#151102) 2025-05-01 18:55:57 +00:00
_device.py Remove torch functions that do not support device arguments from _device_constructor (#150290) 2025-04-08 15:13:55 +00:00
_dtype_abbrs.py [BE] Migrate dtype_abbrs into one location (#152229) 2025-04-28 03:52:47 +00:00
_exposed_in.py
_filelock.py
_foreach_utils.py
_freeze.py
_functools.py
_get_clean_triton.py
_import_utils.py
_mode_utils.py
_ordered_set.py
_python_dispatch.py
_pytree.py [BE] detect CXX pytree requirement with TorchVersion (#151102) 2025-05-01 18:55:57 +00:00
_stats.py
_thunk.py
_traceback.py
_triton.py Revert "has_triton: Use the device interface for detecting Triton availability (#139171)" 2025-05-10 14:46:23 +00:00
_typing_utils.py
_zip.py
backend_registration.py
bundled_inputs.py
checkpoint.py [ca][dynamo] always run eager checkpoint region's recomputation in eager (#153300) 2025-05-16 01:37:48 +00:00
collect_env.py [BE]: Enable ruff YTT linter for Python version checks (#153547) 2025-05-14 21:09:16 +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 Add __all__ for torch.utils.dlpack (#149026) 2025-04-11 22:03:24 +00:00
file_baton.py Warn user of existing lock file to avoid infinite waiting (#149382) 2025-04-15 20:25:29 +00:00
flop_counter.py
hooks.py
mkldnn.py
mobile_optimizer.py
model_zoo.py
module_tracker.py
show_pickle.py
throughput_benchmark.py
weak.py