pytorch

_awaits

…

_C

[Memento] Add PT2 to Memory Snapshot (#152707 )

2025-05-12 21:12:51 +00:00

_C_flatbuffer

…

_custom_op

PEP585 update - mostly toplevels (#145178 )

2025-01-22 02:21:14 +00:00

_decomp

Fix torch.isin decomposition for scalar inputs (#153216 )

2025-05-09 20:26:25 +00:00

_dispatch

[BE][PYFMT] migrate PYFMT for torch._dynamo to ruff format (#144549 )

2025-02-28 03:03:53 +00:00

_dynamo

[dynamo][compile-time] Cache the function signature to speedup inlining (#153396 )

2025-05-14 14:01:46 +00:00

_export

[export] support functools.partial forward (non-strict) (#153408 )

2025-05-13 23:30:13 +00:00

_functorch

[compile-time traces] Profile large missing gaps in compile time (#151256 )

2025-05-13 14:44:51 +00:00

_higher_order_ops

Revert "[export][cond] support merging constant ints as unbacked symint (#152742 )"

2025-05-12 23:06:33 +00:00

_inductor

[AOTI][reland2] Remove typedef for half and bfloat16 (#153467 )

2025-05-14 02:37:18 +00:00

_lazy

PEP585 update - torch/_C torch/_decomp torch/_lazy torch/_library torch/_numpy torch/_prims torch/_refs torch/_strobelight (#145102 )

2025-01-18 20:47:12 +00:00

_library

Add torch._C.Tag.needs_contiguous_strides (#152859 )

2025-05-08 04:49:59 +00:00

_logging

[export] Beef up guard_added logs (#149465 )

2025-03-20 23:02:07 +00:00

_numpy

Fix broken URLs (#152237 )

2025-04-27 09:56:42 +00:00

_prims

Support torch.compile rng selective activation checkpointing with cudagraph (#146878 )

2025-02-28 00:47:03 +00:00

_prims_common

consolidate guard_or_x and definitely_x (#152463 )

2025-05-02 18:08:11 +00:00

_refs

[dynamic shapes] use try-catch instead of guard_or_true for reshape_view_helper (#152638 )

2025-05-06 00:54:24 +00:00

_strobelight

Enable strobelight profiling specific compile frame ids using COMPILE_STROBELIGHT_FRAME_FILTER (#147549 )

2025-02-22 03:44:53 +00:00

_subclasses

[BE]: Update ruff to 0.11.8 (#153249 )

2025-05-12 18:30:52 +00:00

_vendor

…

accelerator

Add torch.accelerator.device_index as accelerator's device switch context (#148864 )

2025-04-25 09:45:25 +00:00

amp

[Intel GPU] skip a cuda api call in amp to save some host overhead on xpu (#151111 )

2025-04-13 06:37:07 +00:00

ao

Fix more URLs (#153277 )

2025-05-14 16:23:50 +00:00

autograd

[autograd][docs] Add more details on why save_for_backward is important in extending autograd note (#153005 )

2025-05-09 16:36:57 +00:00

backends

Revert "refine fp32 precision api (#125888 )"

2025-05-11 00:35:46 +00:00

compiler

[MegaCache] Return None on no compilation (#151921 )

2025-04-23 04:32:06 +00:00

contrib

PEP585 update - mostly toplevels (#145178 )

2025-01-22 02:21:14 +00:00

cpu

[device_mesh] improve device selection logic (#150897 )

2025-05-14 06:29:16 +00:00

csrc

Revert "Rewrite autograd producer consumer stream sync logic (#151079 )"

2025-05-14 13:07:12 +00:00

cuda

make use_mem_pool threadlocal (#153356 )

2025-05-13 00:16:07 +00:00

distributed

Fix negative dim issue in for parallel loss context manager (#152785 )

2025-05-14 10:43:27 +00:00

distributions

[Docs] Add Description of validate_args for torch.distributions (#152173 )

2025-04-30 18:01:20 +00:00

export

[export] Make draft_export public (#153219 )

2025-05-14 02:18:36 +00:00

fft

[BE][Easy] enable PYFMT for torch/[a-s]*/ (#138447 )

2024-12-23 14:04:00 +00:00

func

Add torch.func.debug_unwrap (#146528 )

2025-02-06 18:48:09 +00:00

futures

PEP585: More UP006 fixes (#146392 )

2025-02-20 06:18:13 +00:00

fx

Add skip_dtype_check_in_meta_registrations config to torch/fx/experimental/_config (#153513 )

2025-05-14 09:14:11 +00:00

jit

Fix broken URLs (#152237 )

2025-04-27 09:56:42 +00:00

legacy

…

lib

[1/N] Use internal linkage in torch/csrc C++ files. (#150930 )

2025-04-11 02:19:31 +00:00

linalg

Fix broken URLs (#152237 )

2025-04-27 09:56:42 +00:00

masked

[BE]: Update ruff to 0.11.8 (#153249 )

2025-05-12 18:30:52 +00:00

monitor

add WaitCounter type interface and get rid of type errors (#146175 )

2025-02-01 23:24:52 +00:00

mps

[MPS] Make torch.mps.compile_shader public (#148972 )

2025-03-11 20:20:58 +00:00

mtia

[Kineto] Enable OOM observer (#152160 )

2025-04-27 15:56:44 +00:00

multiprocessing

[BE][CI] bump ruff to 0.8.4 (#143753 )

2024-12-24 12:24:10 +00:00

nativert

[nativert] Move Placement to pytorch core (#152953 )

2025-05-14 15:26:54 +00:00

nested

[aotd] Guess tangents stride as output strides (#144579 )

2025-03-20 15:41:36 +00:00

nn

[BE]: Update ruff to 0.11.8 (#153249 )

2025-05-12 18:30:52 +00:00

onnx

[BE]: Update ruff to 0.11.8 (#153249 )

2025-05-12 18:30:52 +00:00

optim

[BE]: Improve decorator typing for Optimizer subclasses (#153374 )

2025-05-12 22:55:25 +00:00

package

Remove code for Python < 3.9 (#147097 )

2025-02-14 03:22:49 +00:00

profiler

[profiler][retry] don't disable CUPTI_LAZY_REINIT for cuda >= 12.6 (#151124 )

2025-04-15 16:11:49 +00:00

quantization

…

signal

PEP585 update - mostly toplevels (#145178 )

2025-01-22 02:21:14 +00:00

sparse

Revert "has_triton: Use the device interface for detecting Triton availability (#139171 )"

2025-05-10 14:46:23 +00:00

special

[BE][Easy] enable PYFMT for torch/[a-s]*/ (#138447 )

2024-12-23 14:04:00 +00:00

testing

torch.tensordot: performance improvements when contracting to a scalar. (#145936 )

2025-05-13 10:57:30 +00:00

utils

Fix more URLs (#153277 )

2025-05-14 16:23:50 +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

remove allow-untyped-defs for torch/__config__.py (#143320 )

2024-12-17 00:16:09 +00:00

__future__.py

…

__init__.py

Detect NVSHMEM location (#153010 )

2025-05-07 23:35:04 +00:00

_appdirs.py

Fix broken URLs (#152237 )

2025-04-27 09:56:42 +00:00

_classes.py

…

_compile.py

[BE] typing for decorators (#144161 )

2025-01-04 16:40:09 +00:00

_custom_ops.py

…

_deploy.py

…

_environment.py

…

_guards.py

[dynamo][compile-time] Cache the cleaned insturctions while inlining (#153333 )

2025-05-14 09:26:26 +00:00

_jit_internal.py

[BE][CI] bump ruff to 0.9.2: multiline assert statements (#144546 )

2025-02-27 20:46:16 +00:00

_linalg_utils.py

PEP585 update - mostly toplevels (#145178 )

2025-01-22 02:21:14 +00:00

_lobpcg.py

Fixed rerr computation in lobpcg (#152789 )

2025-05-08 12:22:31 +00:00

_lowrank.py

PEP585 update - mostly toplevels (#145178 )

2025-01-22 02:21:14 +00:00

_meta_registrations.py

Add skip_dtype_check_in_meta_registrations config to torch/fx/experimental/_config (#153513 )

2025-05-14 09:14:11 +00:00

_namedtensor_internals.py

…

_ops.py

Introduce unsafe way to mark functions as cacheable (#151603 )

2025-04-21 17:37:38 +00:00

_python_dispatcher.py

…

_size_docs.py

remove allow-untyped-defs from torch/_size_docs.py (#143942 )

2024-12-29 01:00:46 +00:00

_sources.py

PEP585 update - mostly toplevels (#145178 )

2025-01-22 02:21:14 +00:00

_storage_docs.py

…

_streambase.py

…

_tensor_docs.py

Fix broken URLs (#152237 )

2025-04-27 09:56:42 +00:00

_tensor_str.py

add torch.float4_e2m1fn_x2 to PyTorch (#148791 )

2025-03-27 17:32:20 +00:00

_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

Fix the basic description of torch.min(), torch.max(), torch.all(), torch.any() (#152658 )

2025-05-08 22:59:14 +00:00

_utils_internal.py

[reland] Add graph module runtime asserts to AOTI (#153182 )

2025-05-09 22:56:19 +00:00

_utils.py

Allow torch.load under FakeTensorMode to load FakeTensors with correct devices (for plain Tensors) (#147786 )

2025-03-06 12:04:32 +00:00

_VF.py

…

_vmap_internals.py

Fix broken URLs (#152237 )

2025-04-27 09:56:42 +00:00

_weights_only_unpickler.py

Add sparse tensors constructed via legacy constructor to _sparse_tensors_to_validate (#147759 )

2025-02-25 23:51:12 +00:00

CMakeLists.txt

Add new dependences for gen_pyi.py (#150391 )

2025-04-03 14:18:18 +00:00

custom_class_detail.h

Enable readability-redundant-declaration (#143982 )

2024-12-31 00:20:10 +00:00

custom_class.h

Remove unneeded Clang-tidy suppression (#148246 )

2025-03-01 16:51:54 +00:00

extension.h

…

functional.py

Optimize cdist param description (#151178 )

2025-04-14 13:53:10 +00:00

hub.py

[BE][CI][Easy] bump ruff to 0.9.0: long statements in docstrings (#146509 )

2025-02-24 19:56:08 +00:00

library.h

Overload Library::def rather than templating it (#151626 )

2025-04-18 22:51:16 +00:00

library.py

fix spammy library deinit errors when user passes an invalid TORCH_LOGS argument (#151678 )

2025-04-22 20:13:52 +00:00

overrides.py

[CUDA][cuBLAS] Aten GEMM overload for FP32 output from FP16/BF16 inputs (#150812 )

2025-04-18 01:53:26 +00:00

py.typed

…

quasirandom.py

…

random.py

Update description for torch.random.fork_rng (#151881 )

2025-04-23 16:59:29 +00:00

return_types.py

…

script.h

…

serialization.py

Move get accelerator to use build time flags when possible (#146098 )

2025-03-10 13:17:58 +00:00

storage.py

add torch.float4_e2m1fn_x2 to PyTorch (#148791 )

2025-03-27 17:32:20 +00:00

torch_version.py

[BE]: Enable ruff SLOT checks (#146276 )

2025-02-04 19:18:23 +00:00

types.py

Improve typing in torch/types.py (#145237 )

2025-01-28 05:29:12 +00:00

version.py.tpl

…