| .. |
|
_awaits
|
|
|
|
_C
|
Resubmit Remove MemPoolContext (#154042) (#154746)
|
2025-05-31 01:21:54 +00:00 |
|
_C_flatbuffer
|
|
|
|
_custom_op
|
|
|
|
_decomp
|
Fix clamp type promotion in inductor decomposition (#154471)
|
2025-05-28 23:24:25 +00:00 |
|
_dispatch
|
|
|
|
_dynamo
|
[dynamo][guards] Prevent LENGTH guard on nn modules (#154763)
|
2025-05-31 05:32:31 +00:00 |
|
_export
|
[Export] Add math module for deserialization (#154643)
|
2025-05-30 17:29:25 +00:00 |
|
_functorch
|
[BE][Ez]: Optimize unnecessary lambda with operator (#154722)
|
2025-05-30 23:47:10 +00:00 |
|
_higher_order_ops
|
[refactor] move materialize_as_graph to _higher_order_ops/utils.py (#154070)
|
2025-05-31 00:06:44 +00:00 |
|
_inductor
|
[inductor] Add kernel_hash_key to ChoiceCaller (#154470)
|
2025-05-31 03:09:37 +00:00 |
|
_lazy
|
|
|
|
_library
|
Add torch._C.Tag.needs_contiguous_strides (#152859)
|
2025-05-08 04:49:59 +00:00 |
|
_logging
|
Revert "convert inductor codecache to use getArtifactLogger (#153766)"
|
2025-05-30 22:20:07 +00:00 |
|
_numpy
|
|
|
|
_prims
|
[dynamic shapes] unbacked safe unsqueeze (#154087)
|
2025-05-30 01:41:57 +00:00 |
|
_prims_common
|
avoid sym_max on nested int in is_contiguous. (#154633)
|
2025-05-30 09:59:33 +00:00 |
|
_refs
|
introduce definitely_contiguous and use it for reshape and tensor meta data computation. (#153432)
|
2025-05-28 03:41:26 +00:00 |
|
_strobelight
|
|
|
|
_subclasses
|
Revert "[draft export] avoid storing intermediate real tensors in proxies (#154630)"
|
2025-05-31 00:07:56 +00:00 |
|
_vendor
|
|
|
|
accelerator
|
|
|
|
amp
|
|
|
|
ao
|
remove allow-untyped-defs from torch/ao/quantization/stubs.py (#154622)
|
2025-05-30 07:26:09 +00:00 |
|
autograd
|
Add memory reporting for XPU to Memory Profiler (#152842)
|
2025-05-21 01:19:19 +00:00 |
|
backends
|
Revert "refine fp32 precision api (#125888)"
|
2025-05-11 00:35:46 +00:00 |
|
compiler
|
Support unbacked whitelist (#154295)
|
2025-05-28 23:01:22 +00:00 |
|
contrib
|
|
|
|
cpu
|
[device_mesh] improve device selection logic (#150897)
|
2025-05-14 06:29:16 +00:00 |
|
csrc
|
Resubmit Remove MemPoolContext (#154042) (#154746)
|
2025-05-31 01:21:54 +00:00 |
|
cuda
|
Resubmit Remove MemPoolContext (#154042) (#154746)
|
2025-05-31 01:21:54 +00:00 |
|
distributed
|
[BE][Ez]: Optimize unnecessary lambda with operator (#154722)
|
2025-05-30 23:47:10 +00:00 |
|
distributions
|
Type hints for distributions/utils (#154712)
|
2025-05-30 15:50:31 +00:00 |
|
export
|
[export] Move PT2ArchiveWriter/Reader to torch/export (#153795)
|
2025-05-23 19:04:36 +00:00 |
|
fft
|
|
|
|
func
|
|
|
|
futures
|
Render Example: and not Example:: in docs (#153978)
|
2025-05-21 01:03:26 +00:00 |
|
fx
|
Revert "[draft export] avoid storing intermediate real tensors in proxies (#154630)"
|
2025-05-31 00:07:56 +00:00 |
|
jit
|
|
|
|
legacy
|
|
|
|
lib
|
Revert "Use 3.27 as the minimum CMake version (#153153)"
|
2025-05-31 02:14:24 +00:00 |
|
linalg
|
|
|
|
masked
|
Add torch.Tensor._make_wrapper_subclass to torch/_C/__init__.pyi (#154022)
|
2025-05-27 14:10:00 +00:00 |
|
monitor
|
|
|
|
mps
|
|
|
|
mtia
|
Add getDeviceProperties api to torch mtia device (#153577)
|
2025-05-27 11:55:58 +00:00 |
|
multiprocessing
|
|
|
|
nativert
|
[nativert] move OpKernelKind enum to torch (#154756)
|
2025-05-31 01:31:29 +00:00 |
|
nested
|
Add torch.Tensor._make_wrapper_subclass to torch/_C/__init__.pyi (#154022)
|
2025-05-27 14:10:00 +00:00 |
|
nn
|
Fix load_state_dict description (#154599)
|
2025-05-30 18:08:59 +00:00 |
|
onnx
|
[BE][Ez]: Optimize unnecessary lambda with operator (#154722)
|
2025-05-30 23:47:10 +00:00 |
|
optim
|
Fix lr_scheduler unexpectedly calls step() when init argument last_epoch is larger than -1 (#149312)
|
2025-05-22 08:42:37 +00:00 |
|
package
|
[BE]: Enable ruff YTT linter for Python version checks (#153547)
|
2025-05-14 21:09:16 +00:00 |
|
profiler
|
|
|
|
quantization
|
|
|
|
signal
|
|
|
|
sparse
|
Add torch.Tensor._make_wrapper_subclass to torch/_C/__init__.pyi (#154022)
|
2025-05-27 14:10:00 +00:00 |
|
special
|
Render Example: and not Example:: in docs (#153978)
|
2025-05-21 01:03:26 +00:00 |
|
testing
|
[Reland][pytorch] Patch the _is_conv_node function (#154473)
|
2025-05-30 00:41:03 +00:00 |
|
utils
|
[BE]Enhance _get_clean_triton.py to auto-generate launch_params if missing (#154666)
|
2025-05-31 19:27:56 +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
|
[Dynamo] Introduce hook receiving list of traced code objects (#153622)
|
2025-05-28 15:40:09 +00:00 |
|
_appdirs.py
|
|
|
|
_classes.py
|
|
|
|
_compile.py
|
|
|
|
_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
|
[Dynamo] Introduce hook receiving list of traced code objects (#153622)
|
2025-05-28 15:40:09 +00:00 |
|
_jit_internal.py
|
BE: Type previously untyped decorators (#154515)
|
2025-05-29 00:36:34 +00:00 |
|
_linalg_utils.py
|
|
|
|
_lobpcg.py
|
Fixed rerr computation in lobpcg (#152789)
|
2025-05-08 12:22:31 +00:00 |
|
_lowrank.py
|
|
|
|
_meta_registrations.py
|
[Intel GPU][Inductor] Fallback embedding_dense_backward on XPU (#151637)
|
2025-05-19 02:19:37 +00:00 |
|
_namedtensor_internals.py
|
|
|
|
_ops.py
|
Revert "Improve torch.ops typing (#153558)"
|
2025-05-19 23:32:36 +00:00 |
|
_python_dispatcher.py
|
|
|
|
_size_docs.py
|
Render Example: and not Example:: in docs (#153978)
|
2025-05-21 01:03:26 +00:00 |
|
_sources.py
|
|
|
|
_storage_docs.py
|
|
|
|
_streambase.py
|
|
|
|
_tensor_docs.py
|
Render Example: and not Example:: in docs (#153978)
|
2025-05-21 01:03:26 +00:00 |
|
_tensor_str.py
|
|
|
|
_tensor.py
|
|
|
|
_thread_safe_fork.py
|
|
|
|
_torch_docs.py
|
Render Example: and not Example:: in docs (#153978)
|
2025-05-21 01:03:26 +00:00 |
|
_utils_internal.py
|
[reland] Add graph module runtime asserts to AOTI (#153182)
|
2025-05-09 22:56:19 +00:00 |
|
_utils.py
|
Temporarily disable sparse tensor validation when loading from external storage. (#154758)
|
2025-05-31 19:45:44 +00:00 |
|
_VF.py
|
|
|
|
_vmap_internals.py
|
|
|
|
_weights_only_unpickler.py
|
|
|
|
CMakeLists.txt
|
Revert "Use 3.27 as the minimum CMake version (#153153)"
|
2025-05-31 02:14:24 +00:00 |
|
custom_class_detail.h
|
|
|
|
custom_class.h
|
|
|
|
extension.h
|
|
|
|
functional.py
|
|
|
|
header_only_apis.txt
|
Add torch/header_only_apis.txt and enforce they're tested (#153635)
|
2025-05-20 23:42:24 +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
|
|
|
|
py.typed
|
|
|
|
quasirandom.py
|
|
|
|
random.py
|
|
|
|
return_types.py
|
|
|
|
script.h
|
|
|
|
serialization.py
|
Update serialization docs (#153631)
|
2025-05-19 20:22:07 +00:00 |
|
storage.py
|
|
|
|
torch_version.py
|
|
|
|
types.py
|
|
|
|
version.py.tpl
|
|
|