pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00

Files

dolpm 66f53889d5 [nativert] port semaphore to c10 util (#153504 )

Summary:
nativert RFC: https://github.com/zhxchen17/rfcs/blob/master/RFC-0043-torch-native-runtime.md

To land the runtime into PyTorch core, we will gradually land logical parts of the code into the Github issue and get each piece properly reviewed.

This diff adds a simple semaphore interface into c10 until c++20 where we get counting_semaphore

gonna need a oss build export to take a look at this...

Test Plan: CI

Differential Revision: D73882656

Pull Request resolved: https://github.com/pytorch/pytorch/pull/153504
Approved by: https://github.com/zhxchen17

2025-05-28 19:17:30 +00:00

_awaits

…

Remove MemPoolContext (#154042 )

2025-05-28 16:35:48 +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] updating gb_type names for uniqueness (#154452 )

2025-05-28 16:54:10 +00:00

_export

pyfmt lint torch/_export/passes/replace_view_ops_with_view_copy_ops_pass.py (#154488 )

2025-05-28 17:07:15 +00:00

_functorch

Don't CSE unbacked nodes (#154387 )

2025-05-28 02:21:56 +00:00

_higher_order_ops

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

2025-05-22 17:25:38 +00:00

_inductor

[AOTI][cutlass backend] Do not remove the cutlass kernel .o file after packaging (#154155 )

2025-05-28 17:35:19 +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

Add flag _metrics_log_runtime to disable runtime metric logging by default (#153506 )

2025-05-22 01:02:11 +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

use definitely_contiguous for _prim_elementwise_meta short circuit (#153441 )

2025-05-28 03:41:26 +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

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

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

Add deprecation warning for torch.ao.quantization (#153892 )

2025-05-28 16:25:30 +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

add sticky cache pgo (#154418 )

2025-05-27 16:40:18 +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 "Fix the Problems About Defining Static Variable in Inline Function (#147095 )"

2025-05-28 18:22:39 +00:00

cuda

Remove MemPoolContext (#154042 )

2025-05-28 16:35:48 +00:00

distributed

[PP] Allow unused kwargs in ZB path (#153498 )

2025-05-28 13:34:04 +00:00

distributions

Use property instead of ClassVar for Uniform.arg_constraints and Wishart.arg_constraints (#154361 )

2025-05-26 17:48:28 +00:00

export

[export] Move PT2ArchiveWriter/Reader to torch/export (#153795 )

2025-05-23 19:04: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

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

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

[ez] add docblock for ShapeGuardPythonPrinter (#154403 )

2025-05-28 14:17:17 +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

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

[nativert] Move file_util to pytorch core (#153162 )

2025-05-27 03:42:47 +00:00

nested

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

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

[BE][CI][Easy] Run lintrunner on generated .pyi stub files (#150732 )

2025-05-27 14:58:02 +00:00

onnx

[ONNX] Update onnx to 1.18 (#153746 )

2025-05-25 20:58:47 +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

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

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

special

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

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

testing

[BE][CI][Easy] Run lintrunner on generated .pyi stub files (#150732 )

2025-05-27 14:58:02 +00:00

utils

pymft lint torch/utils/weak.py (#154484 )

2025-05-28 17:06:58 +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

Revert "[BE]: Type previously untyped decorators (#153726 )"

2025-05-22 16:49:08 +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

[Intel GPU][Inductor] Fallback embedding_dense_backward on XPU (#151637 )

2025-05-19 02:19:37 +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

…

_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

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

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

_utils.py

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

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

[nativert] port semaphore to c10 util (#153504 )

2025-05-28 19:17:30 +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

Add torch/header_only_apis.txt and enforce they're tested (#153635 )

2025-05-20 23:42:24 +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

…