Animesh Jain 8f3d7972ad [dynamo][compile-time] Cache the function signature to speedup inlining (#153396) ... Pull Request resolved: https://github.com/pytorch/pytorch/pull/153396 Approved by: https://github.com/jansel, https://github.com/StrongerXi ghstack dependencies: #153333		2025-05-14 14:01:46 +00:00
..
_awaits
_C	[Memento] Add PT2 to Memory Snapshot (#152707)	2025-05-12 21:12:51 +00:00
_C_flatbuffer
_custom_op
_decomp	Fix torch.isin decomposition for scalar inputs (#153216)	2025-05-09 20:26:25 +00:00
_dispatch
_dynamo	[dynamo][compile-time] Cache the function signature to speedup inlining (#153396)	2025-05-14 14:01:46 +00:00
_export	[export] support functools.partial forward (non-strict) (#153408)	2025-05-13 23:30:13 +00:00
_functorch	[compile-time traces] Profile large missing gaps in compile time (#151256)	2025-05-13 14:44:51 +00:00
_higher_order_ops	Revert "[export][cond] support merging constant ints as unbacked symint (#152742)"	2025-05-12 23:06:33 +00:00
_inductor	[AOTI][reland2] Remove typedef for half and bfloat16 (#153467)	2025-05-14 02:37:18 +00:00
_lazy
_library	Add torch._C.Tag.needs_contiguous_strides (#152859)	2025-05-08 04:49:59 +00:00
_logging
_numpy	Fix broken URLs (#152237)	2025-04-27 09:56:42 +00:00
_prims
_prims_common	consolidate guard_or_x and definitely_x (#152463)	2025-05-02 18:08:11 +00:00
_refs	[dynamic shapes] use try-catch instead of guard_or_true for reshape_view_helper (#152638)	2025-05-06 00:54:24 +00:00
_strobelight
_subclasses	[BE]: Update ruff to 0.11.8 (#153249)	2025-05-12 18:30:52 +00:00
_vendor
accelerator	Add torch.accelerator.device_index as accelerator's device switch context (#148864)	2025-04-25 09:45:25 +00:00
amp
ao	[BE]: Update ruff to 0.11.8 (#153249)	2025-05-12 18:30:52 +00:00
autograd	[autograd][docs] Add more details on why save_for_backward is important in extending autograd note (#153005)	2025-05-09 16:36:57 +00:00
backends	Revert "refine fp32 precision api (#125888)"	2025-05-11 00:35:46 +00:00
compiler	[MegaCache] Return None on no compilation (#151921)	2025-04-23 04:32:06 +00:00
contrib
cpu	[device_mesh] improve device selection logic (#150897)	2025-05-14 06:29:16 +00:00
csrc	Revert "Rewrite autograd producer consumer stream sync logic (#151079)"	2025-05-14 13:07:12 +00:00
cuda	make use_mem_pool threadlocal (#153356)	2025-05-13 00:16:07 +00:00
distributed	Fix negative dim issue in for parallel loss context manager (#152785)	2025-05-14 10:43:27 +00:00
distributions	[Docs] Add Description of validate_args for torch.distributions (#152173)	2025-04-30 18:01:20 +00:00
export	[export] Make draft_export public (#153219)	2025-05-14 02:18:36 +00:00
fft
func
futures
fx	Add skip_dtype_check_in_meta_registrations config to torch/fx/experimental/_config (#153513)	2025-05-14 09:14:11 +00:00
jit	Fix broken URLs (#152237)	2025-04-27 09:56:42 +00:00
legacy
lib
linalg	Fix broken URLs (#152237)	2025-04-27 09:56:42 +00:00
masked	[BE]: Update ruff to 0.11.8 (#153249)	2025-05-12 18:30:52 +00:00
monitor
mps
mtia	[Kineto] Enable OOM observer (#152160)	2025-04-27 15:56:44 +00:00
multiprocessing
nativert	[nativert] move executor config to torch (#153087)	2025-05-13 23:26:00 +00:00
nested
nn	[BE]: Update ruff to 0.11.8 (#153249)	2025-05-12 18:30:52 +00:00
onnx	[BE]: Update ruff to 0.11.8 (#153249)	2025-05-12 18:30:52 +00:00
optim	[BE]: Improve decorator typing for Optimizer subclasses (#153374)	2025-05-12 22:55:25 +00:00
package
profiler
quantization
signal
sparse	Revert "has_triton: Use the device interface for detecting Triton availability (#139171)"	2025-05-10 14:46:23 +00:00
special
testing	torch.tensordot: performance improvements when contracting to a scalar. (#145936)	2025-05-13 10:57:30 +00:00
utils	Enable ruff check for torch/utils/data/*.ipynb (#148654)	2025-05-14 06:21:47 +00:00
xpu	Correct torch.xpu.is_bf16_supported return False if no XPU detected (#152317)	2025-05-06 10:03:17 +00:00
__config__.py
__future__.py
__init__.py	Detect NVSHMEM location (#153010)	2025-05-07 23:35:04 +00:00
_appdirs.py	Fix broken URLs (#152237)	2025-04-27 09:56:42 +00:00
_classes.py
_compile.py
_custom_ops.py
_deploy.py
_environment.py
_guards.py	[dynamo][compile-time] Cache the cleaned insturctions while inlining (#153333)	2025-05-14 09:26:26 +00:00
_jit_internal.py
_linalg_utils.py
_lobpcg.py	Fixed rerr computation in lobpcg (#152789)	2025-05-08 12:22:31 +00:00
_lowrank.py
_meta_registrations.py	Add skip_dtype_check_in_meta_registrations config to torch/fx/experimental/_config (#153513)	2025-05-14 09:14:11 +00:00
_namedtensor_internals.py
_ops.py	Introduce unsafe way to mark functions as cacheable (#151603)	2025-04-21 17:37:38 +00:00
_python_dispatcher.py
_size_docs.py
_sources.py
_storage_docs.py
_streambase.py
_tensor_docs.py	Fix broken URLs (#152237)	2025-04-27 09:56:42 +00:00
_tensor_str.py
_tensor.py	Avoid triggering ignored requires_grad warning in our code (#152686)	2025-05-05 23:56:40 +00:00
_thread_safe_fork.py
_torch_docs.py	Fix the basic description of torch.min(), torch.max(), torch.all(), torch.any() (#152658)	2025-05-08 22:59:14 +00:00
_utils_internal.py	[reland] Add graph module runtime asserts to AOTI (#153182)	2025-05-09 22:56:19 +00:00
_utils.py
_VF.py
_vmap_internals.py	Fix broken URLs (#152237)	2025-04-27 09:56:42 +00:00
_weights_only_unpickler.py
CMakeLists.txt
custom_class_detail.h
custom_class.h
extension.h
functional.py
hub.py
library.h	Overload Library::def rather than templating it (#151626)	2025-04-18 22:51:16 +00:00
library.py	fix spammy library deinit errors when user passes an invalid TORCH_LOGS argument (#151678)	2025-04-22 20:13:52 +00:00
overrides.py	[CUDA][cuBLAS] Aten GEMM overload for FP32 output from FP16/BF16 inputs (#150812)	2025-04-18 01:53:26 +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
storage.py
torch_version.py
types.py
version.py.tpl