pytorch

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

Files

Yukio Siraichi a10f15718d [DLPack] Add support for missing keyword-arguments. (#150218 )

This PR introduces the rest of the keyword-arguments added in DLPack
version 2023.12: `dl_device` and `copy`.

In summary, we handle these arguments in the C++ implementation of
`to_dlpack(...)` at _torch/csrc/Module.cpp_, by calling the
`maybeCopyTensor` function at _aten/src/ATen/DLConvertor.cpp_. It also
introduces the following changes:

- Add a new Python API `torchDeviceToDLDevice()`, which is simply a
  refactoring of the `getDLDevice()` function at
  _aten/src/ATen/DLConvertor.cpp_.
- Add both keyword-arguments to the `from_dlpack()` function at
  _torch/utils/dlpack.py_ and to the `Tensor.__dlpack__()` dunder
  method.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/150218
Approved by: https://github.com/albanD
ghstack dependencies: #150216, #150217

2025-07-20 00:46:20 +00:00

_awaits

…

[DLPack] Add support for missing keyword-arguments. (#150218 )

2025-07-20 00:46:20 +00:00

_C_flatbuffer

…

_custom_op

pyfmt lint torch/_custom_op/* (#155782 )

2025-06-12 23:04:11 +00:00

_decomp

Fused RMSNorm implementation (#153666 )

2025-07-18 23:24:21 +00:00

_dispatch

Improve torch.ops typing (#154555 )

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

_dynamo

[dynamo] Support more basic output types for nonstrict_trace (#157969 )

2025-07-19 00:59:54 +00:00

_export

Add FP8 Types (#158430 )

2025-07-17 18:09:56 +00:00

_functorch

Revert "Add torch compile force disable caches alias (#158072 )"

2025-07-18 04:58:24 +00:00

_higher_order_ops

[hop] set capture_scalar_outputs=True by default for compiled hops (#158480 )

2025-07-18 07:16:50 +00:00

_inductor

[async-TP] Turn asserts back into silent skips (#158572 )

2025-07-19 04:54:38 +00:00

_lazy

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

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

_library

[BE] remove torch deploy - conditionals (#158288 )

2025-07-17 05:56:07 +00:00

_logging

[dynamo][be] hide warnings without invalidating warnings cache (#158520 )

2025-07-18 22:02:31 +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

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

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

_prims_common

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

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

_refs

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

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

_strobelight

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

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

_subclasses

[dynamo][be] hide warnings without invalidating warnings cache (#158520 )

2025-07-18 22:02:31 +00:00

_vendor

…

accelerator

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

2025-07-17 01:56:01 +00:00

amp

Issue warning with reference to user code rather than torch (#155112 )

2025-07-14 05:24:23 +00:00

Use new type statement to fix public API of types (#158487 )

2025-07-17 18:46:44 +00:00

autograd

Fix types in graphs.py (#158192 )

2025-07-15 19:49:38 +00:00

backends

Expose opt_einsum in torch.backends (#157740 )

2025-07-15 17:46:43 +00:00

compiler

Revert "Add torch compile force disable caches alias (#158072 )"

2025-07-18 04:58:24 +00:00

contrib

…

cpu

[device_mesh] improve device selection logic (#150897 )

2025-05-14 06:29:16 +00:00

csrc

[DLPack] Add support for missing keyword-arguments. (#150218 )

2025-07-20 00:46:20 +00:00

cuda

Revert "Add warning about removed sm50 and sm60 arches (#158301 )"

2025-07-19 00:32:04 +00:00

distributed

[async-TP] Turn asserts back into silent skips (#158572 )

2025-07-19 04:54:38 +00:00

distributions

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

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

export

[Dynamo][Better Engineering] Add enhanced typing support to _dynamo/eval_frame.py (#158276 )

2025-07-16 23:31:10 +00:00

fft

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

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

func

Add torch.func.debug_unwrap (#146528 )

2025-02-06 18:48:09 +00:00

futures

Simplify the base classes of _PyFutureMeta (#157757 )

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

[BE][EZ] Minor doc fixes (#158574 )

2025-07-18 10:34:55 -05:00

headeronly

Add STD_TORCH_CHECK to headeronly (#158377 )

2025-07-18 14:35:20 +00:00

jit

[4/n] Remove references to TorchScript in PyTorch docs (#158317 )

2025-07-16 20:01:34 +00:00

legacy

…

lib

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

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

linalg

Fix for ambiguity in linalg.norm()'s ord argument of +2 & -2 (#155148 )

2025-06-04 21:15:20 +00:00

masked

[Pyrefly][Refactor] Replace dict() calls with literal dict syntax for improved readability (#157735 )

2025-07-08 18:10:33 +00:00

monitor

add WaitCounter type interface and get rid of type errors (#146175 )

2025-02-01 23:24:52 +00:00

mps

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

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

mtia

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

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

multiprocessing

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

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

nativert

[nativert] benchmark util (#158678 )

2025-07-20 00:28:09 +00:00

nested

Add check nested_tensor_from_jagged param jagged_dim >= 1 (#157770 )

2025-07-10 00:34:39 +00:00

Add a way to disable compile for debugging flex-attention (#158534 )

2025-07-18 05:33:45 +00:00

onnx

[BE][EZ] Minor doc fixes (#158574 )

2025-07-18 10:34:55 -05:00

optim

Document the rest of the specific optimizer module APIs (#158669 )

2025-07-19 07:27:15 +00:00

package

[BE][Ez]: Update ruff to 0.12.2 (#157937 )

2025-07-11 15:16:20 +00:00

profiler

[BE][Ez]: Autotype torch/profiler with ruff ANN (#157923 )

2025-07-09 22:07:50 +00:00

quantization

[BE][6/16] fix typos in torch/ (#156316 )

2025-06-23 02:57:34 +00:00

signal

[BE][6/16] fix typos in torch/ (#156316 )

2025-06-23 02:57:34 +00:00

sparse

[build] modernize build-frontend: python setup.py develop/install -> [uv ]pip install --no-build-isolation [-e ]. (#156027 )

2025-07-09 11:24:27 +00:00

special

[BE][6/16] fix typos in torch/ (#156316 )

2025-06-23 02:57:34 +00:00

testing

[ca] reset between tests (#158418 )

2025-07-17 20:14:29 +00:00

utils

[DLPack] Add support for missing keyword-arguments. (#150218 )

2025-07-20 00:46:20 +00:00

xpu

[BE][6/16] fix typos in torch/ (#156316 )

2025-06-23 02:57:34 +00:00

__config__.py

…

__future__.py

…

__init__.py

[BE] remove torch deploy - conditionals (#158288 )

2025-07-17 05:56:07 +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

[precompile] Ensure @disable()-ed function won't trigger recompile from precompile bytecode. (#155363 )

2025-06-10 16:13:38 +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

Revert "[dynamo][fsdp] Consistent behavior of int attributes (#157262 )"

2025-07-10 23:11:18 +00:00

_jit_internal.py

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

2025-07-09 11:02:22 +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

Revert "DDE-Free select with unbacked index. (#157605 )"

2025-07-17 16:20:02 +00:00

_namedtensor_internals.py

…

_ops.py

Revert "[Easy] Show some clear error when torch.ops.load_library fails. (#157524 )"

2025-07-19 00:45:31 +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

Documentation Fix: torch.tensor.scatter_ docs (#157929 )

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

_tensor_str.py

fix tensor print behavior for MAIA (#155609 )

2025-06-14 01:04:12 +00:00

_tensor.py

[DLPack] Add support for missing keyword-arguments. (#150218 )

2025-07-20 00:46:20 +00:00

_thread_safe_fork.py

…

_torch_docs.py

Revert "[cuda][cupy] Improve cupy device placement when device is provided (#158320 )"

2025-07-16 19:15:33 +00:00

_utils_internal.py

[BE] remove torch deploy - conditionals (#158288 )

2025-07-17 05:56:07 +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

Add sparse tensors constructed via legacy constructor to _sparse_tensors_to_validate (#147759 )

2025-02-25 23:51:12 +00:00

CMakeLists.txt

Migrate c10/macros/cmake_macros.h.in to torch/headeronly (#158035 )

2025-07-15 19:52:59 +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

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

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

header_only_apis.txt

Add STD_TORCH_CHECK to headeronly (#158377 )

2025-07-18 14:35:20 +00:00

hub.py

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

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

library.h

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

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

library.py

[BE] remove torch deploy - conditionals (#158288 )

2025-07-17 05:56:07 +00:00

overrides.py

[DLPack] Add support for missing keyword-arguments. (#150218 )

2025-07-20 00:46:20 +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

Reduce random reads for offset metadata when calling torch.load under FakeTensorMode (#157931 )

2025-07-17 22:17:52 +00:00

storage.py

mypy 1.16.0 (#155821 )

2025-06-14 18:18:43 +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

…