pytorch

mirror of https://github.com/zebrajr/pytorch.git synced 2026-01-15 12:15:51 +00:00

Files

Will Constable b2be4d24c0 [DTensor] Make ExplicitRedistributeContext strict/non-strict mode (#167370 )

Also support nesting, enable/disable, and make the class use a
thread-local for storage so independent threads do not confuse each
other.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/167370
Approved by: https://github.com/ezyang
ghstack dependencies: #166593

2025-11-11 01:19:16 +00:00

_awaits

…

[MTIAGraph][Pytorch][3/n] Implement mtia_graph python wrapper in pytorch (#166964 )

2025-11-10 23:49:20 +00:00

_C_flatbuffer

…

_custom_op

Fix flake8 B028 warnings (#166224 )

2025-10-26 06:18:55 +00:00

_decomp

[3/N] Use key in dict for existence checks (#167214 )

2025-11-07 02:49:15 +00:00

_dispatch

[3/N] Use 'is' in callable comparisons (#166780 )

2025-11-01 20:23:56 +00:00

_dynamo

[ez] Remove spammy deprecation log (#167470 )

2025-11-10 21:49:23 +00:00

_export

[export, 3.14] handle patching methods with functools.partial correctly in non-strict export (#167396 )

2025-11-10 20:52:05 +00:00

_functorch

Support AC in default partitioner when functionalization is enabled (#166610 )

2025-11-10 23:09:01 +00:00

_higher_order_ops

[HOP][print]Add make_fx for the proxy with graph module print (#166920 )

2025-11-08 00:34:24 +00:00

_inductor

allow sym_stride, and sym_size lowering in inductor to return ints (#167345 )

2025-11-10 23:29:23 +00:00

_lazy

Fix pyrefly ignores 1/n (#166239 )

2025-10-26 00:44:10 +00:00

_library

[3/N] Use key in dict for existence checks (#167214 )

2025-11-07 02:49:15 +00:00

_logging

compile time comm benchmarking (#167100 )

2025-11-07 20:13:37 +00:00

_numpy

[3/N] Use key in dict for existence checks (#167214 )

2025-11-07 02:49:15 +00:00

_prims

Fix pyrefly ignore syntax (#166438 )

2025-10-29 00:02:21 +00:00

_prims_common

[3/N] Use key in dict for existence checks (#167214 )

2025-11-07 02:49:15 +00:00

_refs

[3/N] fix typo in other folders (#166606 )

2025-10-30 10:30:40 +00:00

_strobelight

Fix pyrefly error syntax (2/n) (#166448 )

2025-10-29 00:36:40 +00:00

_subclasses

[dynamo] rename unimplemented_v2 -> unimplemented (#167150 )

2025-11-08 01:49:53 +00:00

_vendor

…

accelerator

Introduce a new API torch.accelerator.get_memory_info (#156812 )

2025-11-10 01:57:39 +00:00

amp

Fix pyrefly error syntax (2/n) (#166448 )

2025-10-29 00:36:40 +00:00

[BE] adding documentation (#167334 )

2025-11-10 14:46:42 +00:00

autograd

Revert "Remove python workaround for ContextDecorator (#167049 )"

2025-11-07 22:32:45 +00:00

backends

Fix pyrefly error syntax (2/n) (#166448 )

2025-10-29 00:36:40 +00:00

compiler

Expose torch.compiler.config.force_disable_caches as a public API (#166699 )

2025-11-06 05:59:05 +00:00

contrib

…

cpu

Fix pyrefly error syntax (2/n) (#166448 )

2025-10-29 00:36:40 +00:00

csrc

Add shim for at::get_num_threads (#167362 )

2025-11-10 22:21:14 +00:00

cuda

[3/N] Use key in dict for existence checks (#167214 )

2025-11-07 02:49:15 +00:00

distributed

[DTensor] Make ExplicitRedistributeContext strict/non-strict mode (#167370 )

2025-11-11 01:19:16 +00:00

distributions

Fix pyrelfy ignore syntax in distributions and ao (#166248 )

2025-10-26 22:13:48 +00:00

export

[3/N] Use key in dict for existence checks (#167214 )

2025-11-07 02:49:15 +00:00

fft

…

func

…

futures

…

move subgraph_has_impure_ops from node.is_impure into const_fold to unblock production (#167443 )

2025-11-10 19:29:54 +00:00

headeronly

Migrate TypeTraits, TypeList, Metaprogramming to torch:: headeronly (#167386 )

2025-11-11 00:02:20 +00:00

jit

Enable ruff UP035 rule (#167307 )

2025-11-09 06:40:03 +00:00

legacy

…

lib

[3/N] Add clang-tidy readability checks (#164692 )

2025-11-03 07:28:21 +00:00

linalg

…

masked

[3/N] Use key in dict for existence checks (#167214 )

2025-11-07 02:49:15 +00:00

monitor

…

mps

…

mtia

[MTIAGraph][Pytorch][3/n] Implement mtia_graph python wrapper in pytorch (#166964 )

2025-11-10 23:49:20 +00:00

multiprocessing

Fix syntax for pyrefly errors (#166496 )

2025-10-29 20:00:25 +00:00

nativert

[Sigmoid][Delta Update][2/N] update delta update api to load original value first before casting to target dtype (#167039 )

2025-11-06 19:31:18 +00:00

nested

[3/N] Add return types of Python functions (#167287 )

2025-11-07 13:50:33 +00:00

Enable ruff UP035 rule (#167307 )

2025-11-09 06:40:03 +00:00

numa

Fix pyrefly errors on main (#166548 )

2025-10-30 16:47:27 +00:00

onnx

Enable ruff UP035 rule (#167307 )

2025-11-09 06:40:03 +00:00

optim

[1/N] Add return types of Python functions (#167162 )

2025-11-06 04:32:14 +00:00

package

[3/N] Use key in dict for existence checks (#167214 )

2025-11-07 02:49:15 +00:00

profiler

[2/N] Use context managers (#167404 )

2025-11-09 13:38:14 +00:00

quantization

Fix syntax for pyrefly errors (#166496 )

2025-10-29 20:00:25 +00:00

signal

Fix syntax for pyrefly errors (#166496 )

2025-10-29 20:00:25 +00:00

sparse

Use context managers (SIM115) (#166928 )

2025-11-08 03:09:16 +00:00

special

…

testing

Revert "Add min/max support for barebones uint types (#166813 )"

2025-11-10 23:25:22 +00:00

utils

[3.14, dataloader] handle forkserver default mp start method in 3.14 (#167387 )

2025-11-10 20:52:05 +00:00

xpu

[xpu][feature] Add XPU support on torch.accelerator.get_memory_info (#162564 )

2025-11-10 05:34:49 +00:00

__config__.py

…

__future__.py

…

__init__.py

[dynamo, 3.14] enable dynamo in 3.14 (#167384 )

2025-11-10 20:52:05 +00:00

_appdirs.py

Use Python 3.10 typing (#148418 )

2025-11-02 16:16:52 +00:00

_classes.py

…

_compile.py

…

_custom_ops.py

…

_environment.py

…

_guards.py

[1/N] Use key in dict for existence checks (#167035 )

2025-11-06 02:25:10 +00:00

_jit_internal.py

Fix pyrefly ignore syntax (#166438 )

2025-10-29 00:02:21 +00:00

_linalg_utils.py

…

_lobpcg.py

Fix pyrefly ignore syntax (#166438 )

2025-10-29 00:02:21 +00:00

_lowrank.py

…

_meta_registrations.py

Fix torch.linalg.eig inductor stride mismatch (#162484 )

2025-11-04 21:06:58 +00:00

_namedtensor_internals.py

[3/N] Use key in dict for existence checks (#167214 )

2025-11-07 02:49:15 +00:00

_ops.py

Use 'is' in callable comparisons (#166624 )

2025-10-30 19:00:09 +00:00

_python_dispatcher.py

…

_size_docs.py

…

_sources.py

…

_storage_docs.py

…

_streambase.py

…

_tensor_docs.py

…

_tensor_str.py

Fix pyrefly error syntax (2/n) (#166448 )

2025-10-29 00:36:40 +00:00

_tensor.py

fix test_type_hints (#163150 )

2025-11-04 00:29:22 +00:00

_thread_safe_fork.py

…

_torch_docs.py

[random] Add generator arg to rand*_like APIs (#166160 )

2025-11-03 19:58:45 +00:00

_utils_internal.py

Fix pyrefly error syntax (2/n) (#166448 )

2025-10-29 00:36:40 +00:00

_utils.py

Fix flake8 B028 warnings (#166224 )

2025-10-26 06:18:55 +00:00

_VF.py

…

_vmap_internals.py

…

_weights_only_unpickler.py

[3/N] Use key in dict for existence checks (#167214 )

2025-11-07 02:49:15 +00:00

CMakeLists.txt

[ROCm] Use a ROCm version string without hash. (#166336 )

2025-10-28 03:53:55 +00:00

custom_class_detail.h

[3/N] Add clang-tidy readability checks (#164692 )

2025-11-03 07:28:21 +00:00

custom_class.h

…

extension.h

…

functional.py

[3/N] fix typo in other folders (#166606 )

2025-10-30 10:30:40 +00:00

header_only_apis.txt

Migrate TypeTraits, TypeList, Metaprogramming to torch:: headeronly (#167386 )

2025-11-11 00:02:20 +00:00

hub.py

Fix flake8 B028 warnings (#166224 )

2025-10-26 06:18:55 +00:00

library.h

[caffe2] Ignore -Wswitch-enum warnings (#166760 )

2025-11-03 09:37:47 +00:00

library.py

Fix syntax for pyrefly errors (#166496 )

2025-10-29 20:00:25 +00:00

overrides.py

[random] Add generator arg to rand*_like APIs (#166160 )

2025-11-03 19:58:45 +00:00

py.typed

…

quasirandom.py

…

random.py

Fix flake8 B028 warnings (#166224 )

2025-10-26 06:18:55 +00:00

return_types.py

…

script.h

…

serialization.py

[1/N] Use key in dict for existence checks (#167035 )

2025-11-06 02:25:10 +00:00

storage.py

Fix pyrefly ignores 1/n (#166239 )

2025-10-26 00:44:10 +00:00

torch_version.py

…

types.py

Enable PLC0414 on ruff (#165828 )

2025-10-22 04:56:52 +00:00

version.py.tpl

…