pytorch/torch/utils
2023-12-19 01:04:58 +00:00
..
_sympy Prefer is_number over is_constant() (#114513) 2023-12-05 16:56:15 +00:00
backcompat
benchmark [BE]: ruff FURB136: replace ternary with min/max (preview) (#114382) 2023-11-22 22:10:01 +00:00
bottleneck
data Re-enable type checking for distributed_c10d.py (#115223) 2023-12-09 11:07:54 +00:00
hipify Revert "[ROCm] add hipblaslt support (#114329)" 2023-12-19 01:04:58 +00:00
jit
model_dump
tensorboard upgrade lintrunner to the lowest supported versions on python 3.12 (#113562) 2023-11-15 18:12:01 +00:00
viz Fix docstring errors in _composable_state.py, remote_device.py, value_ranges.py, utils.py, run.py, rendezvous.py, launch.py, argparse_util.py, __init__.py, _cycles.py (#112953) 2023-11-08 01:13:09 +00:00
__init__.py New swap function (#111747) 2023-12-08 18:49:35 +00:00
_config_module.py Revert "[dynamo] guarded config (#111299)" (#115386) 2023-12-11 19:35:42 +00:00
_config_typing.pyi Revert "[dynamo] guarded config (#111299)" (#115386) 2023-12-11 19:35:42 +00:00
_content_store.py Fix recompilation issue with content store (#113533) 2023-11-13 23:58:13 +00:00
_contextlib.py Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311) 2023-11-15 17:40:04 +00:00
_cpp_extension_versioner.py
_cuda_trace.py
_cxx_pytree.py [pytree] expand tree_map to accept multi-inputs (#115642) 2023-12-14 06:16:42 +00:00
_device.py Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311) 2023-11-15 17:40:04 +00:00
_foreach_utils.py fix docstring issues in torch.utils (#113335) 2023-11-13 19:37:25 +00:00
_freeze.py [Codemod][python/main_function] caffe2: (#113357) 2023-11-15 22:17:31 +00:00
_mode_utils.py
_python_dispatch.py Expand dynamic dims support for traceable subclasses (#114311) 2023-12-05 21:09:25 +00:00
_pytree.py [pytree] expand tree_map to accept multi-inputs (#115642) 2023-12-14 06:16:42 +00:00
_stats.py
_traceback.py Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311) 2023-11-15 17:40:04 +00:00
_triton.py
_typing_utils.py Re-enable type checking for distributed_c10d.py (#115223) 2023-12-09 11:07:54 +00:00
_zip.py [Codemod][python/main_function] caffe2: (#113357) 2023-11-15 22:17:31 +00:00
backend_registration.py fix docstring issues in torch.utils (#113335) 2023-11-13 19:37:25 +00:00
bundled_inputs.py Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311) 2023-11-15 17:40:04 +00:00
checkpoint.py check tensor subclass when using torch.compile + SAC (#115960) 2023-12-18 17:49:06 +00:00
collect_env.py Give users control over packages in torch.utils.collect_env (#112993) 2023-11-28 22:35:25 +00:00
cpp_backtrace.py Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311) 2023-11-15 17:40:04 +00:00
cpp_extension.py Revert "[ROCm] add hipblaslt support (#114329)" 2023-12-19 01:04:58 +00:00
deterministic.py
dlpack.py
file_baton.py Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311) 2023-11-15 17:40:04 +00:00
flop_counter.py fix docstring issues in torch.utils (#113335) 2023-11-13 19:37:25 +00:00
hooks.py Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311) 2023-11-15 17:40:04 +00:00
mkldnn.py Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311) 2023-11-15 17:40:04 +00:00
mobile_optimizer.py Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311) 2023-11-15 17:40:04 +00:00
model_zoo.py
show_pickle.py
throughput_benchmark.py Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311) 2023-11-15 17:40:04 +00:00
weak.py [pytorch][PR] introduce WeakHashRef (#115216) 2023-12-07 17:48:11 +00:00