pytorch/torch/utils
2024-10-03 06:22:55 +00:00
..
_strobelight
_sympy Revert "Simplify find_localzeros (#133325)" 2024-10-01 13:25:03 +00:00
backcompat
benchmark
bottleneck
data
hipify Revert "raw_alloc ignores PYTORCH_NO_CUDA_MEMORY_CACHING (#131114)" 2024-10-03 06:22:55 +00:00
jit
model_dump
tensorboard Param fixes in docstring (#136097) 2024-09-21 18:56:34 +00:00
viz
__init__.py
_backport_slots.py
_config_module.py
_config_typing.pyi
_content_store.py
_contextlib.py
_cpp_extension_versioner.py
_cxx_pytree.py
_device.py Fix DeviceContext bug (#133729) 2024-08-20 07:14:37 +00:00
_exposed_in.py
_foreach_utils.py
_freeze.py
_get_clean_triton.py Fix edge case in inductor triton clean script (#130837) 2024-08-19 23:46:11 +00:00
_import_utils.py
_mode_utils.py
_ordered_set.py [BE] Fix MYPY issues (#133872) 2024-08-20 16:12:04 +00:00
_python_dispatch.py Turn on type-checking in torch.fx.experimental.symbolic_shapes (#136972) 2024-10-01 13:22:10 +00:00
_pytree.py
_stats.py
_thunk.py
_traceback.py
_triton.py Add Triton CPU as an Inductor backend (#133408) 2024-09-30 20:24:52 +00:00
_typing_utils.py
_zip.py
backend_registration.py
bundled_inputs.py
checkpoint.py Param fixes in docstring (#136097) 2024-09-21 18:56:34 +00:00
collect_env.py
cpp_backtrace.py
cpp_extension.py Add abi flags to cpp_extension cache folder (#136890) 2024-09-28 00:49:56 +00:00
deterministic.py
dlpack.py
file_baton.py
flop_counter.py [BE][Ez]: FURB142,FURB92 misc preview fixes (#133880) 2024-08-21 13:54:51 +00:00
hooks.py
mkldnn.py
mobile_optimizer.py
model_zoo.py
module_tracker.py Move module_tracker to logging for confused hierarchy (#134467) 2024-08-26 19:39:08 +00:00
show_pickle.py
throughput_benchmark.py
weak.py [BE]: Update mypy to 1.11.2 (#133816) 2024-09-16 19:44:11 +00:00