pytorch/torch/utils
2025-05-10 14:46:23 +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 Remove conda refs in tools (#152368) 2025-04-29 02:45:47 +00:00
hipify [fbgemm_gpu] Replace C10_CUDA_KERNEL_LAUNCH_CHECK() in the KernelLauncher (#153178) 2025-05-09 17:43:16 +00:00
jit
model_dump
serialization
tensorboard Fix broken URLs (#152237) 2025-04-27 09:56:42 +00:00
viz [Memory Viz] Add Compile Context to Visualizer (#152862) 2025-05-06 23:09:59 +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
_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 SAC: fix recompute tag propagation for ops with list[tensor] inputs (#152195) 2025-05-05 17:21:00 +00:00
collect_env.py Fix broken URLs (#152237) 2025-04-27 09:56:42 +00:00
cpp_backtrace.py
cpp_extension.py xpu: get xpu arch flags at runtime in cpp_extensions (#152192) 2025-05-09 05:43:50 +00:00
deterministic.py
dlpack.py
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