OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 00:20:18 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
08f4535378
pytorch/torch
History
James Wu 08f4535378 Refactor AOTAutogradCacheEntry into AOTAutogradResult (#166656) 
This PR refactors the name AOTAutogradCacheEntry into AOTAutogradResult, and BundledAOTAutogradCacheEntry into BundledAOTAutogradResult. It also moves all coresponding files to a new file, `aot_autograd_result`, which is analogous to `output_code.py` from Inductor.

Having all these be called cache entries made sense when all we used them for was caching. But with AOT compile using BundledAOTAutogradCacheEntry, we want a more generalized naming structure.

This is a no-op change,  and all existing tests should pass.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/166656
Approved by: https://github.com/zhxchen17
ghstack dependencies: #166650
2025-10-31 18:54:09 +00:00
..
_awaits
_C Introduce a new API torch.xpu.get_per_process_memory_fraction (#165511) 2025-10-30 19:30:09 +00:00
_C_flatbuffer
_custom_op Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
_decomp Revert "address DDE in matmul decomp (#166541)" 2025-10-30 18:11:33 +00:00
_dispatch Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
_dynamo Refactor AOTAutogradCacheEntry into AOTAutogradResult (#166656) 2025-10-31 18:54:09 +00:00
_export Revert "[pytree] add treespec_{leaf,tuple,dict} functions for args_spec modification (#160843)" 2025-10-31 18:31:32 +00:00
_functorch Refactor AOTAutogradCacheEntry into AOTAutogradResult (#166656) 2025-10-31 18:54:09 +00:00
_higher_order_ops [1/N] Remove unused loop variables (#166258) 2025-10-30 12:22:25 +00:00
_inductor Add regional aot eager support to AOTAutogradCacheEntry (#166650) 2025-10-31 18:54:09 +00:00
_lazy Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
_library Revert "Make PT2 compile backprop through custom op without autograd key a hard error (#166367)" 2025-10-31 13:44:05 +00:00
_logging Clean up unused Pyrefly suppressions (#166178) 2025-10-25 05:32:21 +00:00
_numpy Enable PLW0127 in ruff (#165851) 2025-10-21 03:30:57 +00:00
_prims Fix pyrefly ignore syntax (#166438) 2025-10-29 00:02:21 +00:00
_prims_common Fix pyrefly ignore syntax (#166438) 2025-10-29 00:02:21 +00:00
_refs [3/N] fix typo in other folders (#166606) 2025-10-30 10:30:40 +00:00
_strobelight Fix pyrefly error syntax (2/n) (#166448) 2025-10-29 00:36:40 +00:00
_subclasses [2/N] Use 'is' in callable comparisons (#166685) 2025-10-31 08:08:07 +00:00
_vendor
accelerator Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
amp Fix pyrefly error syntax (2/n) (#166448) 2025-10-29 00:36:40 +00:00
ao Revert "[pytree] add treespec_{leaf,tuple,dict} functions for args_spec modification (#160843)" 2025-10-31 18:31:32 +00:00
autograd [1/N] Remove unused loop variables (#166258) 2025-10-30 12:22:25 +00:00
backends Fix pyrefly error syntax (2/n) (#166448) 2025-10-29 00:36:40 +00:00
compiler Fix pyrefly error syntax (2/n) (#166448) 2025-10-29 00:36:40 +00:00
contrib
cpu Fix pyrefly error syntax (2/n) (#166448) 2025-10-29 00:36:40 +00:00
csrc Revert "Make PT2 compile backprop through custom op without autograd key a hard error (#166367)" 2025-10-31 13:44:05 +00:00
cuda [ROCm][CUDA] add unit test utility busy_wait_for_flag (#166218) 2025-10-29 22:40:23 +00:00
distributed [2/N] Use 'is' in callable comparisons (#166685) 2025-10-31 08:08:07 +00:00
distributions Fix pyrelfy ignore syntax in distributions and ao (#166248) 2025-10-26 22:13:48 +00:00
export Revert "[pytree] add treespec_{leaf,tuple,dict} functions for args_spec modification (#160843)" 2025-10-31 18:31:32 +00:00
fft
func
futures [5/N] Apply ruff UP035 rule (#164423) 2025-10-02 07:31:11 +00:00
fx Revert "[pytree] add treespec_{leaf,tuple,dict} functions for args_spec modification (#160843)" 2025-10-31 18:31:32 +00:00
headeronly Document LibTorch ABI more, add README to headeronly (#166661) 2025-10-31 17:18:13 +00:00
jit [1/N] Remove unused loop variables (#166258) 2025-10-30 12:22:25 +00:00
legacy
lib [2/N] Mark unused parameters in C++ code (#165121) 2025-10-15 03:04:39 +00:00
linalg Add initial suppressions for pyrefly (#164177) 2025-10-02 20:57:41 +00:00
masked Fix syntax for pyrefly errors (#166496) 2025-10-29 20:00:25 +00:00
monitor
mps Add type annotations to MPS profiler utilities (#163486) 2025-09-27 23:00:53 +00:00
mtia Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
multiprocessing Fix syntax for pyrefly errors (#166496) 2025-10-29 20:00:25 +00:00
nativert [nativert] Downcast triton double arguments to floats (#166620) 2025-10-31 03:52:20 +00:00
nested [2/N] Use 'is' in callable comparisons (#166685) 2025-10-31 08:08:07 +00:00
nn [2/N] Use 'is' in callable comparisons (#166685) 2025-10-31 08:08:07 +00:00
numa Fix pyrefly errors on main (#166548) 2025-10-30 16:47:27 +00:00
onnx Revise deprecation warning for ONNX exporter (#166692) 2025-10-31 17:23:55 +00:00
optim Revert "[2/N] Add strict parameter to Python zip calls (#166257)" 2025-10-30 13:20:00 +00:00
package Fix syntax for pyrefly errors (#166496) 2025-10-29 20:00:25 +00:00
profiler Revert "[2/N] Add strict parameter to Python zip calls (#166257)" 2025-10-30 13:20:00 +00:00
quantization Fix syntax for pyrefly errors (#166496) 2025-10-29 20:00:25 +00:00
signal Fix syntax for pyrefly errors (#166496) 2025-10-29 20:00:25 +00:00
sparse Revert "[2/N] Add strict parameter to Python zip calls (#166257)" 2025-10-30 13:20:00 +00:00
special
testing [ROCm][CI] remove relaxed tolerance for tf32 tests (#166478) 2025-10-31 16:15:42 +00:00
utils Revert "[pytree] add treespec_{leaf,tuple,dict} functions for args_spec modification (#160843)" 2025-10-31 18:31:32 +00:00
xpu Introduce a new API torch.xpu.get_per_process_memory_fraction (#165511) 2025-10-30 19:30:09 +00:00
__config__.py
__future__.py
__init__.py Fix pyrefly ignore syntax (#166438) 2025-10-29 00:02:21 +00:00
_appdirs.py
_classes.py
_compile.py [4/N] Apply ruff UP035 rule to python code (#164206) 2025-10-01 19:05:53 +00:00
_custom_ops.py
_environment.py
_guards.py Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
_jit_internal.py Fix pyrefly ignore syntax (#166438) 2025-10-29 00:02:21 +00:00
_linalg_utils.py
_lobpcg.py Fix pyrefly ignore syntax (#166438) 2025-10-29 00:02:21 +00:00
_lowrank.py
_meta_registrations.py [xpu][feature] Integrate OneDNN SDPA training forward/backward into XPU OVERRIDEABLE Backend (#162454) 2025-10-31 11:20:38 +00:00
_namedtensor_internals.py
_ops.py Use 'is' in callable comparisons (#166624) 2025-10-30 19:00:09 +00:00
_python_dispatcher.py
_size_docs.py
_sources.py
_storage_docs.py
_streambase.py
_tensor_docs.py [reland] Allow setting grad_dtype on leaf tensors (#164751) 2025-10-08 20:23:13 +00:00
_tensor_str.py Fix pyrefly error syntax (2/n) (#166448) 2025-10-29 00:36:40 +00:00
_tensor.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
_thread_safe_fork.py
_torch_docs.py Clarrifying input output angle unit in the docs for trigonometric fun… (#161248) 2025-10-18 11:53:48 +00:00
_utils_internal.py Fix pyrefly error syntax (2/n) (#166448) 2025-10-29 00:36:40 +00:00
_utils.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
_VF.py
_vmap_internals.py [4/N] Apply ruff UP035 rule to python code (#164206) 2025-10-01 19:05:53 +00:00
_weights_only_unpickler.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
CMakeLists.txt [ROCm] Use a ROCm version string without hash. (#166336) 2025-10-28 03:53:55 +00:00
custom_class_detail.h Mark unused parameters in C++ code (#164912) 2025-10-09 06:23:25 +00:00
custom_class.h Mark unused parameters in C++ code (#164912) 2025-10-09 06:23:25 +00:00
extension.h
functional.py [3/N] fix typo in other folders (#166606) 2025-10-30 10:30:40 +00:00
header_only_apis.txt Move toUnderlying to headeronly (#165694) 2025-10-22 05:31:16 +00:00
hub.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
library.h Mark unused parameters in C++ code (#164912) 2025-10-09 06:23:25 +00:00
library.py Fix syntax for pyrefly errors (#166496) 2025-10-29 20:00:25 +00:00
overrides.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
py.typed
quasirandom.py
random.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
return_types.py
script.h
serialization.py Fix syntax for pyrefly errors (#166496) 2025-10-29 20:00:25 +00:00
storage.py Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
torch_version.py
types.py Enable PLC0414 on ruff (#165828) 2025-10-22 04:56:52 +00:00
version.py.tpl