pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-11-16 07:24:54 +08:00

Files

Ke Wen 5c827a4133 [SymmMem] Multi-root tile reduction (#164757 )

Stack from [ghstack](https://github.com/ezyang/ghstack/tree/0.12.0) (oldest at bottom):

Perform multiple tile reductions concurrently, with each tile reduced to a separate root.

- The number of concurrent reductions can be smaller than world size, i.e. roots can be a subset of all ranks. But all ranks are still required to call into this API.

- Currently supports NVLink SHARP scope only.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/164757
Approved by: https://github.com/weifengpy, https://github.com/fegin
ghstack dependencies: #162243

2025-10-08 17:28:00 +00:00

_awaits

…

Revert "[CUDA] Add experimental green context support for SM carveout (#159104 )"

2025-10-07 20:51:22 +00:00

_C_flatbuffer

…

_custom_op

[2/N] Fix ruff warnings (#164460 )

2025-10-04 03:40:32 +00:00

_decomp

remove more (#164753 )

2025-10-08 14:23:38 +00:00

_dispatch

Add initial suppressions for pyrefly (#164177 )

2025-10-02 20:57:41 +00:00

_dynamo

remove more (#164753 )

2025-10-08 14:23:38 +00:00

_export

Pyrefly suppressions 7/n (#164913 )

2025-10-08 07:27:17 +00:00

_functorch

Pyrefly suppressions 6/n (#164877 )

2025-10-08 02:30:57 +00:00

_higher_order_ops

Revert "fix flex attention eager bwd: more rounding (#164317 )"

2025-10-08 14:29:10 +00:00

_inductor

Revert "Limit path search within range (#164581 )"

2025-10-08 14:56:21 +00:00

_lazy

Add initial suppressions for pyrefly (#164177 )

2025-10-02 20:57:41 +00:00

_library

[opaque_obj] Add make_fx tracing support (#163278 )

2025-10-08 09:09:16 +00:00

_logging

Pyrefly suppressions 6/n (#164877 )

2025-10-08 02:30:57 +00:00

_numpy

[2/N] Fix ruff warnings (#164460 )

2025-10-04 03:40:32 +00:00

_prims

Add pyrefly suppressions 2/n (#164513 )

2025-10-03 02:46:13 +00:00

_prims_common

remove more (#164753 )

2025-10-08 14:23:38 +00:00

_refs

Pyrefly suppressions 6/n (#164877 )

2025-10-08 02:30:57 +00:00

_strobelight

Add initial suppressions for pyrefly (#164177 )

2025-10-02 20:57:41 +00:00

_subclasses

remove more (#164753 )

2025-10-08 14:23:38 +00:00

_vendor

…

accelerator

Add unified memory APIs for torch.accelerator (#152932 )

2025-08-08 17:41:22 +00:00

amp

Add initial suppressions for pyrefly (#164177 )

2025-10-02 20:57:41 +00:00

Pyrefly suppressions 6/n (#164877 )

2025-10-08 02:30:57 +00:00

autograd

Pyrefly suppressions 6/n (#164877 )

2025-10-08 02:30:57 +00:00

backends

[2/N] Fix ruff warnings (#164460 )

2025-10-04 03:40:32 +00:00

compiler

Pyrefly suppressions 6/n (#164877 )

2025-10-08 02:30:57 +00:00

contrib

…

cpu

Add initial suppressions for pyrefly (#164177 )

2025-10-02 20:57:41 +00:00

csrc

[SymmMem] Multi-root tile reduction (#164757 )

2025-10-08 17:28:00 +00:00

cuda

Revert "[CUDA] Add experimental green context support for SM carveout (#159104 )"

2025-10-07 20:51:22 +00:00

distributed

[PP]: Optimize memory by early releasing stage inputs' gradients (#164329 )

2025-10-08 16:12:00 +00:00

distributions

Add pyrefly suppressions (#164748 )

2025-10-07 17:31:18 +00:00

export

Pyrefly suppressions 6/n (#164877 )

2025-10-08 02:30:57 +00:00

fft

[BE][PYFMT] migrate PYFMT for torch/[e-n]*/ to ruff format (#144553 )

2025-06-17 08:18:47 +00:00

func

…

futures

[5/N] Apply ruff UP035 rule (#164423 )

2025-10-02 07:31:11 +00:00

Pyrefly suppressions 7/n (#164913 )

2025-10-08 07:27:17 +00:00

headeronly

Move version.h to torch/headeronly (#164381 )

2025-10-07 17:47:30 +00:00

jit

Pyrefly suppressions 6/n (#164877 )

2025-10-08 02:30:57 +00:00

legacy

…

lib

[2/N] Fix cppcoreguidelines-init-variables suppression (#146237 )

2025-06-19 23:26:42 +00:00

linalg

Add initial suppressions for pyrefly (#164177 )

2025-10-02 20:57:41 +00:00

masked

Add pyrefly suppressions (3/n) (#164588 )

2025-10-03 22:03:03 +00:00

monitor

…

mps

Add type annotations to MPS profiler utilities (#163486 )

2025-09-27 23:00:53 +00:00

mtia

Add initial suppressions for pyrefly (#164177 )

2025-10-02 20:57:41 +00:00

multiprocessing

Add initial suppressions for pyrefly (#164177 )

2025-10-02 20:57:41 +00:00

nativert

Add SVE128 ISA (#158932 )

2025-09-29 14:49:19 +00:00

nested

use check_size instead of check_is_size in ops.py (#164668 )

2025-10-08 14:23:38 +00:00

Pyrefly suppressions 7/n (#164913 )

2025-10-08 07:27:17 +00:00

numa

Add initial suppressions for pyrefly (#164177 )

2025-10-02 20:57:41 +00:00

onnx

[BC-Breaking] Remove long-deprecated casting functions from native_functions.yaml (#164641 )

2025-10-08 08:27:58 +00:00

optim

Pyrefly suppressions 6/n (#164877 )

2025-10-08 02:30:57 +00:00

package

Add initial suppressions for pyrefly (#164177 )

2025-10-02 20:57:41 +00:00

profiler

Pyrefly suppressions 6/n (#164877 )

2025-10-08 02:30:57 +00:00

quantization

Add initial suppressions for pyrefly (#164177 )

2025-10-02 20:57:41 +00:00

signal

Add initial suppressions for pyrefly (#164177 )

2025-10-02 20:57:41 +00:00

sparse

Pyrefly suppressions 6/n (#164877 )

2025-10-08 02:30:57 +00:00

special

[BE][PYFMT] migrate PYFMT for torch/[p-z]*/ to ruff format (#144552 )

2025-08-07 00:09:56 +00:00

testing

Pyrefly suppressions 6/n (#164877 )

2025-10-08 02:30:57 +00:00

utils

Pyrefly suppressions 7/n (#164913 )

2025-10-08 07:27:17 +00:00

xpu

Pyrefly suppressions 6/n (#164877 )

2025-10-08 02:30:57 +00:00

__config__.py

…

__future__.py

…

__init__.py

CUDA 13.0 builds fix on Amazon Linux 2023 (#164870 )

2025-10-07 22:52:53 +00:00

_appdirs.py

Fix broken URLs (#152237 )

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

_classes.py

remove allow-untyped-defs from torch/_classes.py (#157231 )

2025-07-08 00:11:52 +00:00

_compile.py

[4/N] Apply ruff UP035 rule to python code (#164206 )

2025-10-01 19:05:53 +00:00

_custom_ops.py

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

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

_environment.py

…

_guards.py

[4/N] Apply ruff UP035 rule to python code (#164206 )

2025-10-01 19:05:53 +00:00

_jit_internal.py

Revert "Enable all SIM rules except disabled ones (#164645 )"

2025-10-05 19:32:21 +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

Add initial suppressions for pyrefly (#164177 )

2025-10-02 20:57:41 +00:00

_lowrank.py

[BE][1/16] fix typos in torch/ (#156311 )

2025-07-09 11:02:22 +00:00

_meta_registrations.py

remove more (#164753 )

2025-10-08 14:23:38 +00:00

_namedtensor_internals.py

…

_ops.py

Revert "Enable all SIM rules except disabled ones (#164645 )"

2025-10-05 19:32:21 +00:00

_python_dispatcher.py

Typo fixes for "overridden" in comments and function names (#155944 )

2025-06-14 03:37:38 +00:00

_size_docs.py

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

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

_sources.py

…

_storage_docs.py

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

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

_streambase.py

…

_tensor_docs.py

Revert "Support setting grad_dtype on leaf tensors (#162815 )"

2025-10-03 23:14:28 +00:00

_tensor_str.py

Pyrefly suppressions 6/n (#164877 )

2025-10-08 02:30:57 +00:00

_tensor.py

Pyrefly suppressions 6/n (#164877 )

2025-10-08 02:30:57 +00:00

_thread_safe_fork.py

…

_torch_docs.py

Update documentation for torch.index_select (#163616 )

2025-09-25 18:29:17 +00:00

_utils_internal.py

Pyrefly suppressions 6/n (#164877 )

2025-10-08 02:30:57 +00:00

_utils.py

Add initial suppressions for pyrefly (#164177 )

2025-10-02 20:57:41 +00:00

_VF.py

…

_vmap_internals.py

[4/N] Apply ruff UP035 rule to python code (#164206 )

2025-10-01 19:05:53 +00:00

_weights_only_unpickler.py

[4/N] Apply ruff UP035 rule to python code (#164206 )

2025-10-01 19:05:53 +00:00

CMakeLists.txt

Revert "[RELAND] Always build USE_DISTRIBUTED (#160449 ) and Make distributed modules importable even when backend not built (#159889 ) (#162594 )"

2025-09-25 13:47:46 +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

[2/N] Fix ruff warnings (#164460 )

2025-10-04 03:40:32 +00:00

header_only_apis.txt

Migrate DeviceType to torch/headeronly (#163999 )

2025-09-30 23:13:27 +00:00

hub.py

Add initial suppressions for pyrefly (#164177 )

2025-10-02 20:57:41 +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 initial suppressions for pyrefly (#164177 )

2025-10-02 20:57:41 +00:00

overrides.py

Revert "Support setting grad_dtype on leaf tensors (#162815 )"

2025-10-03 23:14:28 +00:00

py.typed

…

quasirandom.py

…

random.py

Revert "Add device argument to torch.random.get_rng_state (#163034 )"

2025-10-04 15:25:45 +00:00

return_types.py

…

script.h

…

serialization.py

Add initial suppressions for pyrefly (#164177 )

2025-10-02 20:57:41 +00:00

storage.py

Add initial suppressions for pyrefly (#164177 )

2025-10-02 20:57:41 +00:00

torch_version.py

…

types.py

[4/N] Apply ruff UP035 rule to python code (#164206 )

2025-10-01 19:05:53 +00:00

version.py.tpl

…