OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 37 Packages Projects Releases Wiki Activity
b51d803785
pytorch/torch
History
Yu, Guangye b51d803785 [Dynamo] Add XPU API to trace_rules (#155788) 
# Motivation
- Add binding API and non-bindling API to trace rules for XPU;
- Add some XPU API to the const fold function for Dynamo capture.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/155788
Approved by: https://github.com/jansel, https://github.com/EikanWang
ghstack dependencies: #155787
2025-06-15 17:09:57 +00:00
..
_awaits
_C [BE][c10d/Store]add check in pyi (#155855) (#155865) 2025-06-13 15:39:27 +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 [Dynamo] Add XPU API to trace_rules (#155788) 2025-06-15 17:09:57 +00:00
_export [BE][AOTI] Remove duplicate schema for ExternKernelNode (#155867) 2025-06-14 02:03:27 +00:00
_functorch Typo fixes for "overridden" in comments and function names (#155944) 2025-06-14 03:37:38 +00:00
_higher_order_ops mypy 1.16.0 (#155821) 2025-06-14 18:18:43 +00:00
_inductor [Cutlass] Fix buffer missing issues (#155897) 2025-06-15 00:08:50 +00:00
_lazy
_library Custom Op handle 1-element tuples (#155447) 2025-06-11 03:43:40 +00:00
_logging Migrate from lru_cache to cache (#155613) 2025-06-11 19:44:18 +00:00
_numpy fix numpy compatibility for 2d small list indices (#154806) 2025-06-04 01:58:52 +00:00
_prims Migrate from lru_cache to cache (#155613) 2025-06-11 19:44:18 +00:00
_prims_common Revert "Add view_simple as meta function for view, and avoid calling reshape_view_helper. (#154757)" 2025-06-13 19:11:43 +00:00
_refs Revert "Add view_simple as meta function for view, and avoid calling reshape_view_helper. (#154757)" 2025-06-13 19:11:43 +00:00
_strobelight
_subclasses mypy 1.16.0 (#155821) 2025-06-14 18:18:43 +00:00
_vendor
accelerator
amp
ao mypy 1.16.0 (#155821) 2025-06-14 18:18:43 +00:00
autograd mypy 1.16.0 (#155821) 2025-06-14 18:18:43 +00:00
backends [coreml-backend-tool] fix pytorch-backended issue on new coremltools (#155543) 2025-06-11 20:52:26 +00:00
compiler [precompile] Implement PrecompileContext for recording precompile artifacts, integrate with CompilePackage (#154415) 2025-06-13 14:11:24 +00:00
contrib
cpu [device_mesh] improve device selection logic (#150897) 2025-05-14 06:29:16 +00:00
csrc [BE][Ez]: Optimize nvshmem alloc with missing move (#156000) 2025-06-15 03:04:08 +00:00
cuda [Memory Snapshot] Add Flag to Toggle Global and Local Callbacks for Annotations (#154932) 2025-06-04 23:15:19 +00:00
distributed mypy 1.16.0 (#155821) 2025-06-14 18:18:43 +00:00
distributions Type hints for distributions/utils (#154712) 2025-05-30 15:50:31 +00:00
export mypy 1.16.0 (#155821) 2025-06-14 18:18:43 +00:00
fft
func
futures Render Example: and not Example:: in docs (#153978) 2025-05-21 01:03:26 +00:00
fx mypy 1.16.0 (#155821) 2025-06-14 18:18:43 +00:00
jit
legacy
lib [Reland] Use 3.27 as the minimum CMake version (#154783) 2025-06-14 16:37:51 +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 mypy 1.16.0 (#155821) 2025-06-14 18:18:43 +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 execution frame to torch (#155830) 2025-06-14 03:28:55 +00:00
nested mypy 1.16.0 (#155821) 2025-06-14 18:18:43 +00:00
nn mypy 1.16.0 (#155821) 2025-06-14 18:18:43 +00:00
onnx mypy 1.16.0 (#155821) 2025-06-14 18:18:43 +00:00
optim mypy 1.16.0 (#155821) 2025-06-14 18:18:43 +00:00
package [BE]: Enable ruff YTT linter for Python version checks (#153547) 2025-05-14 21:09:16 +00:00
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 Allow MultiProcContinuousTest to set world_size (#155920) 2025-06-15 00:24:17 +00:00
utils Add__int__ and __float__ methods to _sympy.functions.Identity (#155873) 2025-06-15 04:24:40 +00:00
xpu
__config__.py
__future__.py
__init__.py Replace frame_traced_fn hook with get_traced_code() util (#155249) 2025-06-10 22:40:58 +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 Revert "Add view_simple as meta function for view, and avoid calling reshape_view_helper. (#154757)" 2025-06-13 19:11:43 +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 Optimize Tensor.backward type hints (#155656) 2025-06-13 19:16:48 +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 User-controlled sparse tensor validation when loading data from external storage (#154610) 2025-06-02 10:17:07 +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 Typo fixes for "overridden" in comments and function names (#155944) 2025-06-14 03:37:38 +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