pytorch/torch
cyy 08dae945ae Use official CUDAToolkit module in CMake (#154595)
Use CUDA language in CMake and remove forked FindCUDAToolkit.cmake.
Some CUDA targets are also renamed with `torch::` prefix.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/154595
Approved by: https://github.com/albanD
2025-06-22 05:44:29 +00:00
..
_awaits
_C Enables NCCL symmetric memory kernels through mempool registration (#155134) 2025-06-21 23:24:04 +00:00
_C_flatbuffer
_custom_op pyfmt lint torch/_custom_op/* (#155782) 2025-06-12 23:04:11 +00:00
_decomp fix slice w/ dynamic shapes (#153131) 2025-06-13 15:53:17 +00:00
_dispatch
_dynamo Don't change set_skip_guard_eval_unsafe for DisableContext, since compiler won't run (#156490) 2025-06-22 00:51:32 +00:00
_export [aoti] package weights to disk and dedup (#155241) 2025-06-19 17:17:17 +00:00
_functorch [PT2][partitioners] raise getitems in partitioners to allow earlier release of buffers (#155809) 2025-06-21 19:57:21 +00:00
_higher_order_ops [invoke_subgraph] Make invoke_subgraph cacheable (#156448) 2025-06-20 21:20:23 +00:00
_inductor [logging] dynamo_timed for CachingAutotuner.coordinate_descent_tuning (#156517) 2025-06-21 14:17:19 +00:00
_lazy
_library Custom Op handle 1-element tuples (#155447) 2025-06-11 03:43:40 +00:00
_logging Add logging for async compile worker statistics (#155820) 2025-06-19 23:10:15 +00:00
_numpy fix numpy compatibility for 2d small list indices (#154806) 2025-06-04 01:58:52 +00:00
_prims remove allow-untyped-defs from context.py (#155622) 2025-06-16 07:38:34 +00:00
_prims_common Add view_simple as meta function for view, and avoid calling reshape_view_helper for unbacked (#154757) 2025-06-19 04:50:18 +00:00
_refs remove gso from vector_norm (#156530) 2025-06-21 08:42:36 +00:00
_strobelight
_subclasses remove make_fast_binary_impl from make_fast_binary_impl (#156528) 2025-06-21 08:27:54 +00:00
_vendor
accelerator [Accelerator] Fix Python typing in accelerator (#152394) 2025-06-17 01:27:40 +00:00
amp
ao mypy 1.16.0 (#155821) 2025-06-14 18:18:43 +00:00
autograd [aot][ca] save bw_module in AOTAutogradCache (#151860) 2025-06-19 03:47:41 +00:00
backends [coreml-backend-tool] fix pytorch-backended issue on new coremltools (#155543) 2025-06-11 20:52:26 +00:00
compiler [compile][hierarchical compilation] Release nested_compile_region API (#156449) 2025-06-21 15:14:59 +00:00
contrib
cpu
csrc use guard_or_false for expand utils reduction (#155868) 2025-06-21 23:42:19 +00:00
cuda Enables NCCL symmetric memory kernels through mempool registration (#155134) 2025-06-21 23:24:04 +00:00
distributed [SymmMem] Add NVSHMEM Quiet support to Triton (#156475) 2025-06-21 22:19:58 +00:00
distributions Type hints for distributions/utils (#154712) 2025-05-30 15:50:31 +00:00
export [aoti] package weights to disk and dedup (#155241) 2025-06-19 17:17:17 +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 [BE][PYFMT] migrate PYFMT for torch/[e-n]*/ to ruff format (#144553) 2025-06-17 08:18:47 +00:00
fx s/defer_runtime_assert/guard_or_defer_runtime_assert (#156397) 2025-06-19 10:18:28 +00:00
jit [BE][PYFMT] migrate PYFMT for torch/[e-n]*/ to ruff format (#144553) 2025-06-17 08:18:47 +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 [BE][PYFMT] migrate PYFMT for torch/[e-n]*/ to ruff format (#144553) 2025-06-17 08:18:47 +00:00
monitor
mps [BE][PYFMT] migrate PYFMT for torch/[e-n]*/ to ruff format (#144553) 2025-06-17 08:18:47 +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][PYFMT] migrate PYFMT for torch/[e-n]*/ to ruff format (#144553) 2025-06-17 08:18:47 +00:00
nativert Revert "[nativert] move layout planner algorithms to libtorch (#156508)" 2025-06-21 13:42:40 +00:00
nested mypy 1.16.0 (#155821) 2025-06-14 18:18:43 +00:00
nn fix hack to check if register_buffer has been overridden (#155963) 2025-06-18 01:50:30 +00:00
onnx [ONNX] Implement Attention-23 (#156431) 2025-06-20 23:54:57 +00:00
optim mypy 1.16.0 (#155821) 2025-06-14 18:18:43 +00:00
package
profiler Typo fixes for "overridden" in comments and function names (#155944) 2025-06-14 03:37:38 +00:00
quantization
signal
sparse mypy 1.16.0 (#155821) 2025-06-14 18:18:43 +00:00
special Add doc for missing functions for torch.special module (#155074) 2025-06-09 22:28:26 +00:00
standalone/macros Move c10/macros/Export.h to torch/standalone (#154850) 2025-06-03 06:18:59 +00:00
testing [ca] skip on some PYTORCH_TEST_WITH_DYNAMO=1 autograd tests (#156374) 2025-06-21 18:33:38 +00:00
utils Use official CUDAToolkit module in CMake (#154595) 2025-06-22 05:44:29 +00:00
xpu
__config__.py
__future__.py
__init__.py [dynamo] Add documentation for guard_filter_fn (#156114) 2025-06-19 16:13:12 +00:00
_appdirs.py
_classes.py
_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
_deploy.py
_environment.py
_guards.py Replace frame_traced_fn hook with get_traced_code() util (#155249) 2025-06-10 22:40:58 +00:00
_jit_internal.py BE: Type previously untyped decorators (#154515) 2025-05-29 00:36:34 +00:00
_linalg_utils.py
_lobpcg.py
_lowrank.py
_meta_registrations.py Refine alignment check along dynamic dimension for grouped MMs (#155466) 2025-06-20 19:42:57 +00:00
_namedtensor_internals.py
_ops.py [hop] auto functionalize invoke_subgraph (#154072) 2025-06-11 22:52:28 +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 [docs] Add docstring indicating UB for converting inf to int (#154781) 2025-06-10 14:04:50 +00:00
_tensor_str.py fix tensor print behavior for MAIA (#155609) 2025-06-14 01:04:12 +00:00
_tensor.py Revert "Upgrade to DLPack 1.0. (#145000)" 2025-06-20 15:32:47 +00:00
_thread_safe_fork.py
_torch_docs.py Update gradient behavior note in torch.amin and torch.amax (#155071) 2025-06-15 16:09:31 +00:00
_utils_internal.py Migrate from lru_cache to cache (#155613) 2025-06-11 19:44:18 +00:00
_utils.py Disable pinning check when loading sparse tensors (#154638) 2025-06-18 14:33:36 +00:00
_VF.py
_vmap_internals.py
_weights_only_unpickler.py
CMakeLists.txt [Reland] Use 3.27 as the minimum CMake version (#154783) 2025-06-14 16:37:51 +00:00
custom_class_detail.h
custom_class.h
extension.h
functional.py
header_only_apis.txt Move c10/macros/Export.h to torch/standalone (#154850) 2025-06-03 06:18:59 +00:00
hub.py
library.h
library.py Render Example: and not Example:: in docs (#153978) 2025-05-21 01:03:26 +00:00
overrides.py Revert "Upgrade to DLPack 1.0. (#145000)" 2025-06-20 15:32:47 +00:00
py.typed
quasirandom.py
random.py
return_types.py
script.h
serialization.py mypy 1.16.0 (#155821) 2025-06-14 18:18:43 +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