pytorch/torch
2025-08-26 12:20:37 +00:00
..
_awaits
_C [1/n][export] Refactor PT2 Archive weight saving and loading (#160394) 2025-08-26 01:15:42 +00:00
_C_flatbuffer
_custom_op [BE]: ruff PLC0207 - use maxsplit kwarg (#160107) 2025-08-08 03:14:59 +00:00
_decomp Revert "[dynamic shapes] unbacked-safe slicing (#157944)" 2025-08-22 20:48:46 +00:00
_dispatch
_dynamo [dynamo, nested graph breaks] prevent excessive recompilations (#159786) 2025-08-26 00:58:38 +00:00
_export [1/n][export] Refactor PT2 Archive weight saving and loading (#160394) 2025-08-26 01:15:42 +00:00
_functorch Revert "Close some sources of fake tensor leakages (#159923)" 2025-08-22 20:42:50 +00:00
_higher_order_ops [invoke_subgraph][inductor] Thread graphsafe rng input states for hops (#160713) 2025-08-21 20:41:29 +00:00
_inductor Revert "[inductor] structured-log graph execution order + test (#160448)" 2025-08-26 12:20:37 +00:00
_lazy [BE][2/16] fix typos in torch/ (torch/_*/) (#156312) 2025-07-12 05:47:06 +00:00
_library Account for triton kernel source code hidden in custom ops properly in AOTAutogradCache (#160120) 2025-08-12 14:11:06 +00:00
_logging Add compile_id: Optional[CompileID] to torch._logging._internal.trace_structured_artifact (#160440) 2025-08-13 06:28:23 +00:00
_numpy Fix torch._numpy to match NumPy when empty ellipsis causes advanced indexing separation (#158297) 2025-07-16 08:11:53 +00:00
_prims [BE]: ruff PLC0207 - use maxsplit kwarg (#160107) 2025-08-08 03:14:59 +00:00
_prims_common [dynamic shapes] prims_common non_overlapping_and_dense (#160462) 2025-08-19 01:35:28 +00:00
_refs Revert "Remove guard_size_oblivious from default contiguity python check, and add aten.sym_is_contiguous. (#159197)" 2025-08-18 07:22:13 +00:00
_strobelight [BE][2/16] fix typos in torch/ (torch/_*/) (#156312) 2025-07-12 05:47:06 +00:00
_subclasses Revert "[dynamic shapes] unbacked-safe slicing (#157944)" 2025-08-22 20:48:46 +00:00
_vendor
accelerator Add unified memory APIs for torch.accelerator (#152932) 2025-08-08 17:41:22 +00:00
amp Fix autocast context manager when there is exception (#159565) 2025-08-01 02:12:24 +00:00
ao Remove the uncessary empty file (#160728) 2025-08-19 10:54:08 +00:00
autograd Add ownership token when needed on GradientEdge (#160098) 2025-08-12 20:14:18 +00:00
backends [ROCm] SDPA fix mem fault when dropout is enabled (#154864) 2025-08-21 14:23:13 +00:00
compiler [PGO] add extra read/write keys (#160715) 2025-08-18 01:41:08 +00:00
contrib
cpu Replace _device_t with torch.types.Device in torch/cpu/__init__.py (#161031) 2025-08-21 00:22:43 +00:00
csrc [cpp_wrapper] Swap to new PyBind11 simple GIL header (#161063) 2025-08-26 01:21:18 +00:00
cuda [CUDAGraph] Add getter for cuda graph exec (#161294) 2025-08-25 20:57:37 +00:00
distributed [BE] fix compute_global_tensor_shape test (#161441) 2025-08-26 03:22:29 +00:00
distributions [BE][1/16] fix typos in torch/ (#156311) 2025-07-09 11:02:22 +00:00
export [1/n][export] Refactor PT2 Archive weight saving and loading (#160394) 2025-08-26 01:15:42 +00:00
fft
func
futures Simplify the base classes of _PyFutureMeta (#157757) 2025-07-08 15:39:56 +00:00
fx [tgif] fix getattr_recursive with ModuleList (#161204) 2025-08-25 10:08:47 +00:00
headeronly [Reland] Migrate ScalarType to headeronly (#159911) 2025-08-06 07:36:37 +00:00
jit [4/n] Remove references to TorchScript in PyTorch docs (#158317) 2025-07-16 20:01:34 +00:00
legacy
lib
linalg
masked Revert "Remove guard_size_oblivious from default contiguity python check, and add aten.sym_is_contiguous. (#159197)" 2025-08-18 07:22:13 +00:00
monitor
mps
mtia [Re-land][Inductor] Support native Inductor as backend for MTIA (#159211) 2025-07-29 17:03:24 +00:00
multiprocessing Support NUMA Binding for Callable Entrypoints, Take 2 (#161183) 2025-08-23 07:23:22 +00:00
nativert [1/n][export] Refactor PT2 Archive weight saving and loading (#160394) 2025-08-26 01:15:42 +00:00
nested Revert "Remove guard_size_oblivious from default contiguity python check, and add aten.sym_is_contiguous. (#159197)" 2025-08-18 07:22:13 +00:00
nn typing debugging.py (#160364) 2025-08-15 02:09:31 +00:00
numa Support NUMA Binding for Callable Entrypoints, Take 2 (#161183) 2025-08-23 07:23:22 +00:00
onnx [ONNX] Fix lower opset version support in dynamo=True (#161056) 2025-08-23 05:04:36 +00:00
optim [small][muon] Use addmm for Newton–Schulz orthogonalization (#161379) 2025-08-26 09:17:28 +00:00
package [BE][PYFMT] migrate PYFMT for torch/[p-z]*/ to ruff format (#144552) 2025-08-07 00:09:56 +00:00
profiler [BE][PYFMT] migrate PYFMT for torch/[p-z]*/ to ruff format (#144552) 2025-08-07 00:09:56 +00:00
quantization [BE][PYFMT] migrate PYFMT for torch/[p-z]*/ to ruff format (#144552) 2025-08-07 00:09:56 +00:00
signal [BE][PYFMT] migrate PYFMT for torch/[p-z]*/ to ruff format (#144552) 2025-08-07 00:09:56 +00:00
sparse [BE][PYFMT] migrate PYFMT for torch/[p-z]*/ to ruff format (#144552) 2025-08-07 00:09:56 +00:00
special [BE][PYFMT] migrate PYFMT for torch/[p-z]*/ to ruff format (#144552) 2025-08-07 00:09:56 +00:00
testing [small][muon] Use addmm for Newton–Schulz orthogonalization (#161379) 2025-08-26 09:17:28 +00:00
utils Update pybind11 submodule to 3.0.1 (#160754) 2025-08-26 01:21:18 +00:00
xpu [BE][PYFMT] migrate PYFMT for torch/[p-z]*/ to ruff format (#144552) 2025-08-07 00:09:56 +00:00
__config__.py
__future__.py
__init__.py space added between type and checking for typechecking (#161352) 2025-08-26 02:07:33 +00:00
_appdirs.py
_classes.py remove allow-untyped-defs from torch/_classes.py (#157231) 2025-07-08 00:11:52 +00:00
_compile.py
_custom_ops.py
_environment.py
_guards.py Move save guard error throwing to separate phase (#160662) 2025-08-19 14:46:43 +00:00
_jit_internal.py [BE][1/16] fix typos in torch/ (#156311) 2025-07-09 11:02:22 +00:00
_linalg_utils.py Update is_sparse doc to mention that it is sparse_coo specific (#157378) 2025-07-09 18:22:14 +00:00
_lobpcg.py [BE][1/16] fix typos in torch/ (#156311) 2025-07-09 11:02:22 +00:00
_lowrank.py [BE][1/16] fix typos in torch/ (#156311) 2025-07-09 11:02:22 +00:00
_meta_registrations.py Add meta for add.Scalar (#161332) 2025-08-26 02:26:51 +00:00
_namedtensor_internals.py
_ops.py [BE] remove torch deploy - conditionals (#158288) 2025-07-29 17:40:49 +00:00
_python_dispatcher.py
_size_docs.py
_sources.py
_storage_docs.py
_streambase.py
_tensor_docs.py Add missing optional for tensor ops (#159028) 2025-07-25 04:36:55 +00:00
_tensor_str.py Fix max_width computation in _tensor_str._Formatter (#126859) 2025-08-01 15:05:41 +00:00
_tensor.py [MPS] Enable dlpack integration (#158888) 2025-07-24 18:05:41 +00:00
_thread_safe_fork.py
_torch_docs.py [cuda][cupy] Improve cupy device placement when device is provided with explicit index (#158529) 2025-08-15 00:27:42 +00:00
_utils_internal.py Wire in pt2_triton_builds (#159897) 2025-08-06 07:39:51 +00:00
_utils.py [BE][1/16] fix typos in torch/ (#156311) 2025-07-09 11:02:22 +00:00
_VF.py
_vmap_internals.py
_weights_only_unpickler.py added class or module info for functions blocked by weight-only load (#159935) 2025-08-12 20:52:25 +00:00
CMakeLists.txt CMake build: preserve PYTHONPATH (#160144) 2025-08-08 16:03:49 +00:00
custom_class_detail.h
custom_class.h [BE][1/16] fix typos in torch/ (#156311) 2025-07-09 11:02:22 +00:00
extension.h
functional.py unify broadcast_shapes functions and avoid duplicates (#160251) 2025-08-16 00:54:32 +00:00
header_only_apis.txt [Reland] Migrate ScalarType to headeronly (#159911) 2025-08-06 07:36:37 +00:00
hub.py Allow torch.hub.load with unauthorized GITHUB_TOKEN (#159896) 2025-08-14 18:15:49 +00:00
library.h Using std::make_unique<T>() instead of unique<T>(new T()) (#160723) 2025-08-19 10:25:47 +00:00
library.py Add utility to get computed kernel in torch.library (#158393) 2025-08-13 21:00:59 +00:00
overrides.py [PT2]: Add Static Dispatch Kernel for wrapped_fbgemm_linear_fp16_weight (#160451) 2025-08-15 04:06:17 +00:00
py.typed
quasirandom.py
random.py
return_types.py
script.h
serialization.py added class or module info for functions blocked by weight-only load (#159935) 2025-08-12 20:52:25 +00:00
storage.py
torch_version.py
types.py
version.py.tpl