pytorch

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

Files

IvanKobzarev f85e238186 [aotd] capture rrelu_with_noise noise mutation in compile (#141867 )

Rebase-copy of long standing already approved PR https://github.com/pytorch/pytorch/pull/138503 that was blocked on landing by xla build issues.

Got a new  PR with the same content (ghstack checkout was failing due to changed submodules)

Corresponding xla PR:
https://github.com/pytorch/xla/pull/8363

Pull Request resolved: https://github.com/pytorch/pytorch/pull/141867
Approved by: https://github.com/bdhirsh

2024-12-04 12:18:58 +00:00

_awaits

Remove unused imported names in python files (#134438 )

2024-08-27 20:44:04 +00:00

Add API query for available per-process CUDA memory (#140620 )

2024-12-03 00:24:03 +00:00

_C_flatbuffer

…

_custom_op

Tighten torch.library.infer_schema input types (#130705 )

2024-07-29 16:01:19 +00:00

_decomp

[aotd] capture rrelu_with_noise noise mutation in compile (#141867 )

2024-12-04 12:18:58 +00:00

_dispatch

[BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767 )

2024-07-31 21:18:11 +00:00

_dynamo

[Redo] Set remote cache version and backend type once in compilation metrics (#141967 )

2024-12-04 03:07:53 +00:00

_export

[BE] Remove unused argument (#141983 )

2024-12-04 00:07:33 +00:00

_functorch

Revert "[REFACTOR] Inline FxGraphCache.post_compile into sole call site (#141877 )"

2024-12-03 04:57:58 +00:00

_higher_order_ops

Make torch.library.triton_op public (#141880 )

2024-12-03 16:28:56 +00:00

_inductor

[aotd] capture rrelu_with_noise noise mutation in compile (#141867 )

2024-12-04 12:18:58 +00:00

_lazy

Remove unused imported names in python files (#134438 )

2024-08-27 20:44:04 +00:00

_library

Improve torch.library.opcheck and register_autograd docs (#141883 )

2024-12-03 16:28:56 +00:00

_logging

[BE]: Apply PERF401 autofixes from ruff (#140980 )

2024-11-20 17:52:07 +00:00

_numpy

[test/torch_np] Fix usages of deprecated NumPy 2.0 APIs in numpy_tests (#131909 )

2024-08-05 16:21:08 +00:00

_prims

[BE]: Apply PERF401 autofixes from ruff (#140980 )

2024-11-20 17:52:07 +00:00

_prims_common

[BE]: Update mypy to 1.13.0 (#140808 )

2024-12-03 02:50:10 +00:00

_refs

[BE]: Update mypy to 1.13.0 (#140808 )

2024-12-03 02:50:10 +00:00

_strobelight

Increase default COMPILE_STROBELIGHT_MAX_STACK_LENGTH to 500 (#138006 )

2024-10-17 07:31:32 +00:00

_subclasses

Allow Fakified subclass to have different device for inner and outer tensor (#141839 )

2024-12-03 00:09:41 +00:00

_vendor

…

accelerator

Introduce a device-agnostic runtime API design (#132204 )

2024-10-27 10:37:09 +00:00

amp

[MPS] Add support for bf16 autocast (#139390 )

2024-11-20 19:52:28 +00:00

Add option to split Linear gates for Quantizable LSTM into separate ops (#141366 )

2024-12-03 17:21:44 -05:00

autograd

Missing space in torch.autograd.Function deprecation warning (#141562 )

2024-11-27 01:31:26 +00:00

backends

Revert "[sparse] add search for optimal alg_id to torch.compile (#137427 )"

2024-10-24 17:27:06 +00:00

compiler

[dynamo] skip_guard_eval_unsafe stance for power users (#140251 )

2024-11-21 06:28:58 +00:00

contrib

Remove unused Python variables in torch/[b-z]* (#136963 )

2024-10-19 16:45:22 +00:00

cpu

[Inductor][CPP] Add oneDNN BRGEMM config for Half cpp gemm template (#136255 )

2024-11-05 05:33:29 +00:00

csrc

Enforce contiguity for alltoall (#141816 )

2024-12-04 10:17:39 +00:00

cuda

Add API query for available per-process CUDA memory (#140620 )

2024-12-03 00:24:03 +00:00

distributed

[pipelining] Improve schedule csv loading (#142009 )

2024-12-04 04:15:34 +00:00

distributions

[BE] Use torch.special.expm1 (#141518 )

2024-11-26 01:47:11 +00:00

export

improve typings around torch.export (#141829 )

2024-12-03 19:57:21 +00:00

fft

…

func

…

futures

…

[BE]: Update mypy to 1.13.0 (#140808 )

2024-12-03 02:50:10 +00:00

jit

[BE]: Update mypy to 1.13.0 (#140808 )

2024-12-03 02:50:10 +00:00

legacy

…

lib

Add and use thread-safe strerror (#140472 )

2024-11-19 04:24:17 +00:00

linalg

fix linalg.SVD docs typo: wrong V* shape in reduced SVD (#142037 )

2024-12-04 09:18:33 +00:00

masked

Correctly specify size of sparse_csr tensors in maskedtensor binary ops (#134335 )

2024-12-03 02:55:57 +00:00

monitor

[pytorch][counters] Pybind for WaitCounter (#132357 )

2024-08-02 16:08:10 +00:00

mps

[BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769 )

2024-08-04 10:24:09 +00:00

mtia

[MTIA] Support torch.mtia.empty_cache() (#141533 )

2024-11-28 02:24:19 +00:00

multiprocessing

Remove unused Python variables in torch/[b-z]* (#136963 )

2024-10-19 16:45:22 +00:00

nested

Switch to using Python nested int (#141166 )

2024-12-02 19:17:30 +00:00

Convert assert to torch._check in MHA (#141918 )

2024-12-03 21:58:02 +00:00

onnx

[ONNX] Remove special handling of torchvision.ops imports in onnx export (#141569 )

2024-11-28 18:05:40 +00:00

optim

[BE]: Update mypy to 1.13.0 (#140808 )

2024-12-03 02:50:10 +00:00

package

[BE]: Apply PERF401 autofixes from ruff (#140980 )

2024-11-20 17:52:07 +00:00

profiler

Add skip_first_wait to profiler.schedule (V2) (#141512 )

2024-11-26 18:10:54 +00:00

quantization

Remove unused Python variables in torch/[b-z]* (#136963 )

2024-10-19 16:45:22 +00:00

signal

Fix docstring for torch.signal.windows.nuttall (#134512 )

2024-08-26 22:51:43 +00:00

sparse

[sparse] add extra options to _cslt_spare_mm (#137427 )

2024-11-27 05:32:45 +00:00

special

…

testing

Enable CI on SM89 (#140305 )

2024-12-03 04:49:46 +00:00

utils

When serializing treespec context, support enum as well (#141525 )

2024-12-04 03:08:50 +00:00

xpu

[BE]: Apply PERF401 autofixes from ruff (#140980 )

2024-11-20 17:52:07 +00:00

__config__.py

…

__future__.py

…

__init__.py

[dynamo] add SymNode bitwise and/or (#138777 )

2024-11-22 23:36:16 +00:00

_appdirs.py

[BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765 )

2024-07-31 10:42:50 +00:00

_classes.py

Add None return type to init (#132335 )

2024-08-01 15:26:45 +00:00

_compile.py

[BE] Format uncategorized Python files with ruff format (#132576 )

2024-08-04 17:13:31 +00:00

_custom_ops.py

[BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765 )

2024-07-31 10:42:50 +00:00

_deploy.py

Remove unused imported names in python files (#134438 )

2024-08-27 20:44:04 +00:00

_environment.py

Improve is_fbcode functionality (#136871 )

2024-09-27 21:19:01 +00:00

_guards.py

dynamo: guard on FSDP module parameters (#138819 )

2024-11-13 20:46:46 +00:00

_jit_internal.py

[BE] Format uncategorized Python files with ruff format (#132576 )

2024-08-04 17:13:31 +00:00

_linalg_utils.py

[BE] Format uncategorized Python files with ruff format (#132576 )

2024-08-04 17:13:31 +00:00

_lobpcg.py

[BE][Easy] enable ruff rule PIE790: unnecessary pass statement (#133200 )

2024-08-15 15:50:19 +00:00

_lowrank.py

[BE] Format uncategorized Python files with ruff format (#132576 )

2024-08-04 17:13:31 +00:00

_meta_registrations.py

[aotd] capture rrelu_with_noise noise mutation in compile (#141867 )

2024-12-04 12:18:58 +00:00

_namedtensor_internals.py

[BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765 )

2024-07-31 10:42:50 +00:00

_ops.py

remove redundant a (#139046 )

2024-10-28 17:47:24 +00:00

_python_dispatcher.py

Add None return type to init (#132335 )

2024-08-01 15:26:45 +00:00

_size_docs.py

…

_sources.py

…

_storage_docs.py

[BE] Format uncategorized Python files with ruff format (#132576 )

2024-08-04 17:13:31 +00:00

_streambase.py

Use torch.Stream&torch.Event for Dynamo capature (#134850 )

2024-10-02 14:15:33 +00:00

_tensor_docs.py

Revert "Add deterministic path for CUDA cumsum (#136224 )"

2024-09-27 12:54:47 +00:00

_tensor_str.py

Speed up half tensors printing (#141927 )

2024-12-03 07:01:49 +00:00

_tensor.py

type annotations for meta_utils (#140203 )

2024-11-13 20:07:47 +00:00

_thread_safe_fork.py

[inductor] parallel compile: add import of thread_safe_fork for internal (#137155 )

2024-10-03 17:37:21 +00:00

_torch_docs.py

Clarify torch.arange floating-point rounding behavior (#141655 )

2024-11-27 09:31:39 +00:00

_utils_internal.py

Change export IR to non-functional pre-dispatch IR (#139511 )

2024-11-20 21:47:55 +00:00

_utils.py

[Device] Add mps as device type in torch._utils._get_available_device_type() (#141098 )

2024-11-20 20:45:59 +00:00

_VF.py

Clean up RemoteCache classes (#134032 )

2024-08-31 20:18:59 +00:00

_vmap_internals.py

…

_weights_only_unpickler.py

Add small test case for #140230 (#140850 )

2024-11-19 02:44:54 +00:00

abi-check.cpp

…

CMakeLists.txt

Add torch.version.xpu (#139466 )

2024-11-09 13:31:21 +00:00

custom_class_detail.h

Remove some pre-cpp17 stuff (#138410 )

2024-10-23 00:38:03 +00:00

custom_class.h

Remove some pre-cpp17 stuff (#138410 )

2024-10-23 00:38:03 +00:00

extension.h

…

functional.py

Clarify opt-einsum usage, fix #127109 (#137596 )

2024-10-09 20:31:24 +00:00

hub.py

Remove unused Python variables in torch/[b-z]* (#136963 )

2024-10-19 16:45:22 +00:00

library.h

[1/N] Enable cppcoreguidelines-special-member-functions (#137405 )

2024-10-23 00:16:53 +00:00

library.py

Improve torch.library.opcheck and register_autograd docs (#141883 )

2024-12-03 16:28:56 +00:00

overrides.py

Add Weighted Loss Functions to PyTorch : WMSE, WMAE, and Weighted Huber Loss (#132049 )

2024-10-31 21:59:43 +00:00

py.typed

…

quasirandom.py

[BE] Format uncategorized Python files with ruff format (#132576 )

2024-08-04 17:13:31 +00:00

random.py

[Torch] Support meta device in random.fork_rng (#137715 )

2024-10-16 18:00:39 +00:00

README.txt

…

return_types.py

…

script.h

…

serialization.py

Mention version of flip in weights_only error message (#141304 )

2024-12-03 03:26:27 +00:00

storage.py

Fix .to(cpu) for Storage (#138011 )

2024-10-23 01:31:48 +00:00

torch_version.py

…

types.py

[BE] better type annotation for torch.types (#129559 )

2024-09-02 15:35:32 +00:00

version.py.tpl

…

README.txt

Note [TH abstraction violation]
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

TH/THC provide some hpp headers, which are proper C++ headers rather than
C headers.  These headers serve double duty as *internal implementation
detail* headers, whose contents should largely not be used by external
clients.

Ideally, we would not install these headers at all; instead, you should
use public functions (in headers like `THTensor.h`, NOT `THTensor.hpp`)
to manipulate these structs.  However, there are a few places
in torch/csrc where we violate this abstraction.  They are marked with
a pointer to this note.  Each of those sites will have to be refactored
when we refactor the guts of THTensor and related structures.