pytorch/torch
2025-05-31 00:07:56 +00:00
..
_awaits
_C Revert "Remove MemPoolContext (#154042)" 2025-05-30 06:53:37 +00:00
_C_flatbuffer
_custom_op
_decomp Fix clamp type promotion in inductor decomposition (#154471) 2025-05-28 23:24:25 +00:00
_dispatch
_dynamo [pgo] log dynamic whitelist in PT2 Compile Events (#154747) 2025-05-30 23:54:24 +00:00
_export [Export] Add math module for deserialization (#154643) 2025-05-30 17:29:25 +00:00
_functorch [BE][Ez]: Optimize unnecessary lambda with operator (#154722) 2025-05-30 23:47:10 +00:00
_higher_order_ops [refactor] move materialize_as_graph to _higher_order_ops/utils.py (#154070) 2025-05-31 00:06:44 +00:00
_inductor Use 3.27 as the minimum CMake version (#153153) 2025-05-31 00:01:52 +00:00
_lazy
_library Add torch._C.Tag.needs_contiguous_strides (#152859) 2025-05-08 04:49:59 +00:00
_logging Revert "convert inductor codecache to use getArtifactLogger (#153766)" 2025-05-30 22:20:07 +00:00
_numpy
_prims [dynamic shapes] unbacked safe unsqueeze (#154087) 2025-05-30 01:41:57 +00:00
_prims_common avoid sym_max on nested int in is_contiguous. (#154633) 2025-05-30 09:59:33 +00:00
_refs introduce definitely_contiguous and use it for reshape and tensor meta data computation. (#153432) 2025-05-28 03:41:26 +00:00
_strobelight
_subclasses Revert "[draft export] avoid storing intermediate real tensors in proxies (#154630)" 2025-05-31 00:07:56 +00:00
_vendor
accelerator
amp
ao remove allow-untyped-defs from torch/ao/quantization/stubs.py (#154622) 2025-05-30 07:26:09 +00:00
autograd Add memory reporting for XPU to Memory Profiler (#152842) 2025-05-21 01:19:19 +00:00
backends Revert "refine fp32 precision api (#125888)" 2025-05-11 00:35:46 +00:00
compiler Support unbacked whitelist (#154295) 2025-05-28 23:01:22 +00:00
contrib
cpu [device_mesh] improve device selection logic (#150897) 2025-05-14 06:29:16 +00:00
csrc remove another instance of mtia_workloadd from pytorch (#154739) 2025-05-30 20:50:46 +00:00
cuda Revert "Remove MemPoolContext (#154042)" 2025-05-30 06:53:37 +00:00
distributed [BE][Ez]: Optimize unnecessary lambda with operator (#154722) 2025-05-30 23:47:10 +00:00
distributions Type hints for distributions/utils (#154712) 2025-05-30 15:50:31 +00:00
export [export] Move PT2ArchiveWriter/Reader to torch/export (#153795) 2025-05-23 19:04:36 +00:00
fft
func
futures Render Example: and not Example:: in docs (#153978) 2025-05-21 01:03:26 +00:00
fx Revert "[draft export] avoid storing intermediate real tensors in proxies (#154630)" 2025-05-31 00:07:56 +00:00
jit
legacy
lib Use 3.27 as the minimum CMake version (#153153) 2025-05-31 00:01:52 +00:00
linalg
masked Add torch.Tensor._make_wrapper_subclass to torch/_C/__init__.pyi (#154022) 2025-05-27 14:10:00 +00:00
monitor
mps
mtia Add getDeviceProperties api to torch mtia device (#153577) 2025-05-27 11:55:58 +00:00
multiprocessing
nativert [nativert] move layout planner settings to torch (#154668) 2025-05-30 17:33:27 +00:00
nested Add torch.Tensor._make_wrapper_subclass to torch/_C/__init__.pyi (#154022) 2025-05-27 14:10:00 +00:00
nn Fix load_state_dict description (#154599) 2025-05-30 18:08:59 +00:00
onnx [BE][Ez]: Optimize unnecessary lambda with operator (#154722) 2025-05-30 23:47:10 +00:00
optim Fix lr_scheduler unexpectedly calls step() when init argument last_epoch is larger than -1 (#149312) 2025-05-22 08:42:37 +00:00
package [BE]: Enable ruff YTT linter for Python version checks (#153547) 2025-05-14 21:09:16 +00:00
profiler
quantization
signal
sparse Add torch.Tensor._make_wrapper_subclass to torch/_C/__init__.pyi (#154022) 2025-05-27 14:10:00 +00:00
special Render Example: and not Example:: in docs (#153978) 2025-05-21 01:03:26 +00:00
testing [Reland][pytorch] Patch the _is_conv_node function (#154473) 2025-05-30 00:41:03 +00:00
utils Revert "inductor codecache: include private inductor configs in cache key (#153672)" 2025-05-30 15:54:14 +00:00
xpu Correct torch.xpu.is_bf16_supported return False if no XPU detected (#152317) 2025-05-06 10:03:17 +00:00
__config__.py
__future__.py
__init__.py [Dynamo] Introduce hook receiving list of traced code objects (#153622) 2025-05-28 15:40:09 +00:00
_appdirs.py
_classes.py
_compile.py
_custom_ops.py Render Example: and not Example:: in docs (#153978) 2025-05-21 01:03:26 +00:00
_deploy.py
_environment.py
_guards.py [Dynamo] Introduce hook receiving list of traced code objects (#153622) 2025-05-28 15:40:09 +00:00
_jit_internal.py BE: Type previously untyped decorators (#154515) 2025-05-29 00:36:34 +00:00
_linalg_utils.py
_lobpcg.py Fixed rerr computation in lobpcg (#152789) 2025-05-08 12:22:31 +00:00
_lowrank.py
_meta_registrations.py [Intel GPU][Inductor] Fallback embedding_dense_backward on XPU (#151637) 2025-05-19 02:19:37 +00:00
_namedtensor_internals.py
_ops.py Revert "Improve torch.ops typing (#153558)" 2025-05-19 23:32:36 +00:00
_python_dispatcher.py
_size_docs.py Render Example: and not Example:: in docs (#153978) 2025-05-21 01:03:26 +00:00
_sources.py
_storage_docs.py
_streambase.py
_tensor_docs.py Render Example: and not Example:: in docs (#153978) 2025-05-21 01:03:26 +00:00
_tensor_str.py
_tensor.py Avoid triggering ignored requires_grad warning in our code (#152686) 2025-05-05 23:56:40 +00:00
_thread_safe_fork.py
_torch_docs.py Render Example: and not Example:: in docs (#153978) 2025-05-21 01:03:26 +00:00
_utils_internal.py [reland] Add graph module runtime asserts to AOTI (#153182) 2025-05-09 22:56:19 +00:00
_utils.py Add torch.Tensor._make_wrapper_subclass to torch/_C/__init__.pyi (#154022) 2025-05-27 14:10:00 +00:00
_VF.py
_vmap_internals.py
_weights_only_unpickler.py
CMakeLists.txt Use 3.27 as the minimum CMake version (#153153) 2025-05-31 00:01:52 +00:00
custom_class_detail.h
custom_class.h
extension.h
functional.py
header_only_apis.txt Add torch/header_only_apis.txt and enforce they're tested (#153635) 2025-05-20 23:42:24 +00:00
hub.py
library.h
library.py Render Example: and not Example:: in docs (#153978) 2025-05-21 01:03:26 +00:00
overrides.py
py.typed
quasirandom.py
random.py
return_types.py
script.h
serialization.py Update serialization docs (#153631) 2025-05-19 20:22:07 +00:00
storage.py
torch_version.py
types.py
version.py.tpl