pytorch

_awaits

…

_C

[export] Refactor pt2 save/load (#152495 )

2025-06-04 06:04:29 +00:00

_C_flatbuffer

…

_custom_op

PEP585 update - mostly toplevels (#145178 )

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

_decomp

Fix clamp type promotion in inductor decomposition (#154471 )

2025-05-28 23:24:25 +00:00

_dispatch

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

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

_dynamo

[reland][dynamo] Record the pre-graph bytecode using fast record function event (#154974 )

2025-06-06 13:11:03 +00:00

_export

[Export] Add math module for deserialization (#154643 )

2025-05-30 17:29:25 +00:00

_functorch

Redo D75092426: [internal] Expose additional metadata to compilation callbacks (#155063 )

2025-06-05 23:40:31 +00:00

_higher_order_ops

[scan] disable functionalization key in backward tracing (#154343 )

2025-06-05 20:06:33 +00:00

_inductor

Custom FX pass for inductor's backend registration (#154841 )

2025-06-06 06:49:44 +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

[Tiling rewrite pt1] Normalize reads and writes to common iter space (#153723 )

2025-06-03 14:04:34 +00:00

_numpy

fix numpy compatibility for 2d small list indices (#154806 )

2025-06-04 01:58:52 +00:00

_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

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

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

_subclasses

Revert "Add randint_like tensor overload for high (#154899 )"

2025-06-05 04:54:05 +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 incorrect get_default_qat_qconfig in prepare_qat_fx docs. (#155100 )

2025-06-04 18:51:40 +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

Combine sticky pgo key with job id (#154863 )

2025-06-03 07:58:38 +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

[profiler] Enable all configured activities in CUPTI Range profiler mode (#154749 )

2025-06-05 23:38:54 +00:00

cuda

[Memory Snapshot] Add Flag to Toggle Global and Local Callbacks for Annotations (#154932 )

2025-06-04 23:15:19 +00:00

distributed

Switch to _apply_to_tensors for dataclass input (#154897 )

2025-06-04 02:19:52 +00:00

distributions

Type hints for distributions/utils (#154712 )

2025-05-30 15:50:31 +00:00

export

Adding from_node for nodes in gm.module() (#155053 )

2025-06-05 20:11:56 +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

Render Example: and not Example:: in docs (#153978 )

2025-05-21 01:03:26 +00:00

fx

[typo] Fix 'intialize' -> 'initialize' in proxy_tensor.py (#155301 )

2025-06-06 10:43:44 +00:00

jit

Fix broken URLs (#152237 )

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

legacy

…

lib

Revert "Use 3.27 as the minimum CMake version (#153153 )"

2025-05-31 02:14:24 +00:00

linalg

Fix for ambiguity in linalg.norm()'s ord argument of +2 & -2 (#155148 )

2025-06-04 21:15:20 +00:00

masked

Add torch.Tensor._make_wrapper_subclass to torch/_C/__init__.pyi (#154022 )

2025-05-27 14:10:00 +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

Add getDeviceProperties api to torch mtia device (#153577 )

2025-05-27 11:55:58 +00:00

multiprocessing

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

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

nativert

[2/3] Add source file for Graph in nativert (#154531 )

2025-06-03 21:51:52 +00:00

nested

Add torch.Tensor._make_wrapper_subclass to torch/_C/__init__.pyi (#154022 )

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

nn

Revert "Add pinned numpy and fix build (#155129 )"

2025-06-06 03:14:47 +00:00

onnx

[ONNX] Allow exporter to export SDPA to Attention onnx operator (#154596 )

2025-06-04 14:29:44 +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

[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

fix numpy compatibility for 2d small list indices (#154806 )

2025-06-04 01:58:52 +00:00

special

Render Example: and not Example:: in docs (#153978 )

2025-05-21 01:03:26 +00:00

standalone/macros

Move c10/macros/Export.h to torch/standalone (#154850 )

2025-06-03 06:18:59 +00:00

testing

Custom FX pass for inductor's backend registration (#154841 )

2025-06-06 06:49:44 +00:00

utils

Add Intel GPU info collection to the collect env script (#137846 )

2025-06-06 05:53:24 +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

[Dynamo] Introduce hook receiving list of traced code objects (#153622 )

2025-05-28 15:40:09 +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

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

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

Revert "Add randint_like tensor overload for high (#154899 )"

2025-06-05 04:54:05 +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

PEP585 update - mostly toplevels (#145178 )

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

_storage_docs.py

Fix docstring for torch.UntypedStorage.from_file (#155067 )

2025-06-05 14:30:49 +00:00

_streambase.py

…

_tensor_docs.py

Render Example: and not Example:: in docs (#153978 )

2025-05-21 01:03:26 +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

Render Example: and not Example:: in docs (#153978 )

2025-05-21 01:03:26 +00:00

_utils_internal.py

Revert "Inductor logging + analysis of torch.profile (#149697 )"

2025-06-05 01:38:13 +00:00

_utils.py

User-controlled sparse tensor validation when loading data from external storage (#154610 )

2025-06-02 10:17:07 +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

Move c10/macros/Export.h to torch/standalone (#154850 )

2025-06-03 06:18:59 +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

header_only_apis.txt

Move c10/macros/Export.h to torch/standalone (#154850 )

2025-06-03 06:18:59 +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

Render Example: and not Example:: in docs (#153978 )

2025-05-21 01:03:26 +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

Update serialization docs (#153631 )

2025-05-19 20:22:07 +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

…