OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
35f3572fa4
pytorch/torch
History
anwang bc5111cd8d [Inductor] Prevent kernel fusion with too many unique inputs and outputs (#166275) 
MTIA triton currently has a limit that it can't support the cases when there are too many input/output buffers. This PR adds the limitation to prevent large fusion with many input/output buffer.

Differential Revision: [D85509351](https://our.internmc.facebook.com/intern/diff/D85509351/)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/166275
Approved by: https://github.com/eellison
ghstack dependencies: #166274
2025-10-29 16:41:34 +00:00
..
_awaits
_C Introduce a new API torch.xpu.set_per_process_memory_fraction (#165510) 2025-10-29 03:24:52 +00:00
_C_flatbuffer
_custom_op Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
_decomp Fix pyrefly ignore syntax (#166438) 2025-10-29 00:02:21 +00:00
_dispatch Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
_dynamo Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +00:00
_export [pytree] add treespec_{leaf,tuple,dict} functions for args_spec modification (#160843) 2025-10-29 09:16:24 +00:00
_functorch Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +00:00
_higher_order_ops Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +00:00
_inductor [Inductor] Prevent kernel fusion with too many unique inputs and outputs (#166275) 2025-10-29 16:41:34 +00:00
_lazy Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
_library Fix pyrefly ignore syntax (#166438) 2025-10-29 00:02:21 +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 Fix pyrefly error syntax (2/n) (#166448) 2025-10-29 00:36:40 +00:00
_strobelight Fix pyrefly error syntax (2/n) (#166448) 2025-10-29 00:36:40 +00:00
_subclasses Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +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 "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +00:00
autograd Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +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 "[user-streams] Make device-agnostic streams weakref compatible (#164304)" 2025-10-29 16:09:54 +00:00
cuda filter out alloc-free pairs from trace plot (#165752) 2025-10-29 12:44:54 +00:00
distributed Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +00:00
distributions Fix pyrelfy ignore syntax in distributions and ao (#166248) 2025-10-26 22:13:48 +00:00
export Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +00:00
fft
func
futures
fx add new line in log (#164240) 2025-10-29 16:03:32 +00:00
headeronly Add TORCH_TARGET_VERSION for stable ABI (#164356) 2025-10-29 15:41:28 +00:00
jit Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +00:00
legacy
lib [2/N] Mark unused parameters in C++ code (#165121) 2025-10-15 03:04:39 +00:00
linalg
masked Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
monitor
mps
mtia Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
multiprocessing Replace NUMA inheritance approach (#166026) 2025-10-29 03:58:44 +00:00
nativert [triton][nativert] Add num_cpu_threads for triton-cpu (#166255) 2025-10-28 08:40:04 +00:00
nested Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
nn Revert "bwd pass (#164504)" 2025-10-29 15:10:40 +00:00
numa Replace NUMA inheritance approach (#166026) 2025-10-29 03:58:44 +00:00
onnx Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +00:00
optim Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
package Clean up unused Pyrefly suppressions (#166178) 2025-10-25 05:32:21 +00:00
profiler Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
quantization [RFC] Add pyrefly to lintrunner (#165179) 2025-10-16 20:07:09 +00:00
signal Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
sparse Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
special
testing Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +00:00
utils Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +00:00
xpu Introduce a new API torch.xpu.set_per_process_memory_fraction (#165510) 2025-10-29 03:24:52 +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
_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 [Inductor][Triton][FP8] Support deepseek-style scaling in Inductor (#164404) 2025-10-28 03:38:54 +00:00
_namedtensor_internals.py
_ops.py Fix pyrefly ignore syntax (#166438) 2025-10-29 00:02:21 +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
_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 Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +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 Clean up unused Pyrefly suppressions (#166178) 2025-10-25 05:32:21 +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 Document limitations of weights_only in SECURITY.md and torch.load doc (#165645) 2025-10-27 18:20:50 +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