pytorch/torch/utils
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
..
_strobelight
_sympy Add__int__ and __float__ methods to _sympy.functions.Identity (#155873) 2025-06-15 04:24:40 +00:00
backcompat pyfmt lint more torch/utils files (#155812) 2025-06-12 23:51:42 +00:00
benchmark [BE]: Backport runtime_checkable perf improvements/behavior from 3.12 (#155130) 2025-06-06 13:28:05 +00:00
bottleneck
data Fix #155016 for Docathon - convert rst to markdown (#155198) 2025-06-13 20:24:34 +00:00
hipify support CUBLASLT_MATMUL_MATRIX_SCALE_OUTER_VEC_32F (#154680) 2025-06-18 18:39:01 +00:00
jit
model_dump
serialization
tensorboard
viz [Visualizer] Start at index with most events (#154571) 2025-05-29 20:49:33 +00:00
__init__.py
_appending_byte_serializer.py
_backport_slots.py
_config_module.py Typo fixes for "overridden" in comments and function names (#155944) 2025-06-14 03:37:38 +00:00
_config_typing.pyi inductor codecache: include private inductor configs in cache key (#153672) 2025-06-11 01:33:24 +00:00
_content_store.py
_contextlib.py pyfmt lint more torch/utils files (#155812) 2025-06-12 23:51:42 +00:00
_cpp_embed_headers.py [Build] Allow metal shaders to include ATen headers (#156256) 2025-06-18 01:03:25 +00:00
_cpp_extension_versioner.py pyfmt lint more torch/utils files (#155812) 2025-06-12 23:51:42 +00:00
_cxx_pytree.py
_device.py pyfmt lint more torch/utils files (#155812) 2025-06-12 23:51:42 +00:00
_dtype_abbrs.py
_exposed_in.py
_filelock.py
_foreach_utils.py pyfmt lint more torch/utils files (#155812) 2025-06-12 23:51:42 +00:00
_freeze.py pyfmt lint more torch/utils files (#155812) 2025-06-12 23:51:42 +00:00
_functools.py
_get_clean_triton.py [BE]Enhance _get_clean_triton.py to auto-generate launch_params if missing (#154666) 2025-05-31 19:27:56 +00:00
_helion.py Migrate from lru_cache to cache (#155613) 2025-06-11 19:44:18 +00:00
_import_utils.py
_mode_utils.py pyfmt lint more torch/utils files (#155812) 2025-06-12 23:51:42 +00:00
_ordered_set.py Typo fixes for "overridden" in comments and function names (#155944) 2025-06-14 03:37:38 +00:00
_python_dispatch.py pyfmt lint more torch/utils files (#155812) 2025-06-12 23:51:42 +00:00
_pytree.py Preserve Enum types during torch.export serialization and deserialization (#154821) 2025-06-08 17:30:31 +00:00
_stats.py pyfmt lint more torch/utils files (#155812) 2025-06-12 23:51:42 +00:00
_thunk.py
_traceback.py pyfmt lint more torch/utils files (#155812) 2025-06-12 23:51:42 +00:00
_triton.py [inductor][triton pin] add support for new TMA API for mm.py templates (#155723) 2025-06-12 06:25:47 +00:00
_typing_utils.py
_zip.py pyfmt lint more torch/utils files (#155812) 2025-06-12 23:51:42 +00:00
backend_registration.py pyfmt lint more torch/utils files (#155812) 2025-06-12 23:51:42 +00:00
bundled_inputs.py
checkpoint.py [BE] Mention debug=True in AC error messages (#155593) 2025-06-11 00:32:41 +00:00
collect_env.py Revert "Add Intel GPU info collection to the collect env script (#137846)" 2025-06-11 15:18:47 +00:00
cpp_backtrace.py
cpp_extension.py Use official CUDAToolkit module in CMake (#154595) 2025-06-22 05:44:29 +00:00
deterministic.py
dlpack.py Revert "Upgrade to DLPack 1.0. (#145000)" 2025-06-20 15:32:47 +00:00
file_baton.py
flop_counter.py Revert "Inductor logging + analysis of torch.profile (#149697)" 2025-06-10 15:38:40 +00:00
hooks.py Add warning for module full backward hook when no input requires gradient (#155339) 2025-06-10 04:42:06 +00:00
mkldnn.py
mobile_optimizer.py
model_zoo.py
module_tracker.py
show_pickle.py
throughput_benchmark.py
weak.py