pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00

Files

Isalia20 62b0ebd8f9 [MPS] [Sparse] unique_dim and sparse broadcast (#163694 )

Implements unique_dim, sparse broadcast ops and adds dtypes for mps for tests where we expect to fail, otherwise they would always fail due to being run in double precision

Pull Request resolved: https://github.com/pytorch/pytorch/pull/163694
Approved by: https://github.com/malfet

2025-09-26 23:03:13 +00:00

_awaits

…

Expose Kineto event metadata in PyTorch Profiler events (#161624 )

2025-09-25 14:58:30 +00:00

_C_flatbuffer

…

_custom_op

[BE]: ruff PLC0207 - use maxsplit kwarg (#160107 )

2025-08-08 03:14:59 +00:00

_decomp

[inductor] Fix divmod error in decomp (#163482 )

2025-09-24 02:52:36 +00:00

_dispatch

Improve torch.ops typing (#154555 )

2025-06-22 15:52:27 +00:00

_dynamo

Move inductor.aot_compile to use new tracer (#163137 )

2025-09-26 15:54:24 +00:00

_export

Move inductor.aot_compile to use new tracer (#163137 )

2025-09-26 15:54:24 +00:00

_functorch

Update ruff to 0.13.1 (#163744 )

2025-09-26 10:12:21 +00:00

_higher_order_ops

Fix specialize_impl from triton.runtime.jit (#163844 )

2025-09-26 06:37:26 +00:00

_inductor

[AOTI] Pass in shape_env for get_stride_order (#163925 )

2025-09-26 21:10:03 +00:00

_lazy

[BE][2/16] fix typos in torch/ (torch/_*/) (#156312 )

2025-07-12 05:47:06 +00:00

_library

[ignore][codex-test] Add typing to simple library registry (#161367 )

2025-09-23 02:08:55 +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

[Bugfix] Match eager stride semantics for cloned tensors with preserve_format in compile (#163017 )

2025-09-19 19:41:33 +00:00

_prims_common

are_strides_like_channels_last_or_false (#162354 )

2025-09-16 00:49:05 +00:00

_refs

Better decomp for torch.eye (#163386 )

2025-09-22 21:52:37 +00:00

_strobelight

[BE][2/16] fix typos in torch/ (torch/_*/) (#156312 )

2025-07-12 05:47:06 +00:00

_subclasses

[FakeTensor] Supplement the relevant logic for converting conv1d to conv2d in meta_conv (#160408 )

2025-09-26 15:45:02 +00:00

_vendor

…

accelerator

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

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

amp

[Easy][AMP] Refactor the AMP logic for getting dtype (#162796 )

2025-09-21 06:32:35 +00:00

remove allow-untyped-defs from ./torch/ao/quantization/quantizer/utils.py (#163471 )

2025-09-25 06:48:44 +00:00

autograd

Expose Kineto event metadata in PyTorch Profiler events (#161624 )

2025-09-25 14:58:30 +00:00

backends

Revert "[ROCm] SDPA fix mem fault when dropout is enabled (#154864 )"

2025-08-26 20:03:59 +00:00

compiler

Simplify PrecompileContext to no longer be a CacheArtifactManager (#162886 )

2025-09-20 01:24:37 +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

Add magic TORCH_MAKE_PYBIND_ENUM_FASTER macro (#163527 )

2025-09-26 17:59:22 +00:00

cuda

Revert "[CUDA] Compare major version of the runtime device arch against the built version of the pytorch binary (#161299 )"

2025-09-25 17:13:32 +00:00

distributed

[DeviceMesh][ez] Extract the pg creation as a util function (#163930 )

2025-09-26 20:42:58 +00:00

distributions

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

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

export

Fix preserve annotation with decomp (#163896 )

2025-09-26 20:28:47 +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

Simplify the base classes of _PyFutureMeta (#157757 )

2025-07-08 15:39:56 +00:00

[WIP][precompile] Set fake_mode of base tensor in fx graph pickler (#163738 )

2025-09-26 14:36:37 +00:00

headeronly

Add CUDA_KERNEL_ASSERT_PRINTF, a more flexible CUDA_KERNEL_ASSERT_MSG (#160129 )

2025-09-16 00:23:48 +00:00

jit

[BE] Delete all pre py-3.10 checks (#163653 )

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

legacy

…

lib

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

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

linalg

Revert "Add __init__.pyi to torch/linalg (#160750 )"

2025-09-02 16:53:55 +00:00

masked

Remove guard_size_oblivious from default contiguity python check, and add aten.sym_is_contiguous. [attempt2] (#160869 )

2025-09-08 22:59:13 +00:00

monitor

…

mps

[BE][12/16] fix typos in torch/ (#156602 )

2025-07-02 22:55:29 +00:00

mtia

[BE] Add Documentation for Device APIs (#162834 )

2025-09-16 17:01:06 +00:00

multiprocessing

Allow parallel start NUMA binding (#161576 )

2025-08-28 01:15:58 +00:00

nativert

Add Static Dispatch Kernels (#163676 ) (#163870 )

2025-09-26 03:00:07 +00:00

nested

Add NestedTensor dispatch for _is_any_true/_is_all_true (#162096 )

2025-09-22 20:22:44 +00:00

Avoid fast path mask left-align check in compiled TransformerEncoder (#163773 )

2025-09-26 22:29:37 +00:00

numa

Allow parallel start NUMA binding (#161576 )

2025-08-28 01:15:58 +00:00

onnx

remove allow-untyped-defs from ./torch/onnx/_internal/torchscript_exporter/_globals.py (#163472 )

2025-09-23 03:50:29 +00:00

optim

[optim] prevent unintended aliasing in lr_scheduler; update type annotations/docs (#163120 )

2025-09-25 06:58:58 +00:00

package

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

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

profiler

removed duplicate imports (#161685 )

2025-08-31 16:21:49 +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

Use computed buffer sizes of torch for cusparseLt metadata (#163125 )

2025-09-19 22:12:40 +00:00

special

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

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

testing

[MPS] [Sparse] unique_dim and sparse broadcast (#163694 )

2025-09-26 23:03:13 +00:00

utils

Remove dataclass_slots (#163623 )

2025-09-26 00:54:42 +00:00

xpu

Add a new API torch.xpu.can_device_access_peer for Intel GPU (#162705 )

2025-09-16 18:00:22 +00:00

__config__.py

…

__future__.py

…

__init__.py

[CD] CUDA 13.0 fix preload logic to include nvidia/cu13/lib/ (#163661 )

2025-09-24 11:27:05 +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

Replace Literal[None] with None in typing (#163489 )

2025-09-22 22:10:08 +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

Remove dataclass_slots (#163623 )

2025-09-26 00:54:42 +00:00

_jit_internal.py

[Code Clean] Remove deadcodes about Python3.9 [7/N] (#163646 )

2025-09-24 07:30:50 +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

[FakeTensor] Supplement the relevant logic for converting conv1d to conv2d in meta_conv (#160408 )

2025-09-26 15:45:02 +00:00

_namedtensor_internals.py

…

_ops.py

[BE] Slight improvements to documentation in python_dispatch (#162963 )

2025-09-21 01:45:46 +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

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

Make Tensor.__dlpack__(stream=None) capture-safe during CUDA Graph capture (#163242 )

2025-09-24 16:04:19 +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

Add DISABLE_JUSTKNOBS to torch/_utils_internal.py and use it for dynamo _maybe_set_eval_frame (#162298 )

2025-09-15 23:00:39 +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

Fix broken URLs (#152237 )

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

_weights_only_unpickler.py

Fix type checking for persistent loads in the weights-only unpickler (#161661 )

2025-09-01 19:57:19 +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

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

Replace Literal[None] with None in typing (#163489 )

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

overrides.py

Fully native DTensor.__new__ (#162508 )

2025-09-21 18:36:05 +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

added class or module info for functions blocked by weight-only load (#159935 )

2025-08-12 20:52:25 +00:00

storage.py

mypy 1.16.0 (#155821 )

2025-06-14 18:18:43 +00:00

torch_version.py

…

types.py

…

version.py.tpl

…