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
a25818cf7e
pytorch/torch
History
Nikita Shulga 1abfa5f70b [EZ][MPS] Improve distribution error checking (#166425) 
Essentially not allow ops on self-overlapping outputs, by adding
`at::assert_no_internal_overlap(self);` check that already used in CPU
and CUDA builds, see
895795f07c/aten/src/ATen/native/DistributionTemplates.h (L366)

This fixes `test_error_inputs_bernoulli_mps`

Should be landed ahead of https://github.com/pytorch/pytorch/pull/165267
Pull Request resolved: https://github.com/pytorch/pytorch/pull/166425
Approved by: https://github.com/Skylion007, https://github.com/seemethere
2025-10-28 18:42:12 +00:00
..
_awaits
_C Re-re-re-re-apply "C++-accessible Placements via pybind11 (#163030)" (#166132) 2025-10-27 21:19:32 +00:00
_C_flatbuffer
_custom_op Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
_decomp Back out "Do not decompose in functionalization/proxy tensor if autograd wouldn't have decomposed (#164939)" (#165910) 2025-10-21 16:36:38 +00:00
_dispatch Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
_dynamo docs: fix typos (#164879) 2025-10-28 12:00:36 +00:00
_export docs: fix typos (#164879) 2025-10-28 12:00:36 +00:00
_functorch Export should use aot_export_joint_with_descriptors (#165931) 2025-10-27 19:33:33 +00:00
_higher_order_ops docs: fix typos (#164879) 2025-10-28 12:00:36 +00:00
_inductor refactor: pull _replace_node common functionality out of Scheduler.finalize_multi_template_buffers (#163368) 2025-10-28 17:21:52 +00:00
_lazy Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
_library Clean up unused Pyrefly suppressions (#166178) 2025-10-25 05:32: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 self assignment (#165816) 2025-10-18 18:51:52 +00:00
_prims_common Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
_refs Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
_strobelight Add initial suppressions for pyrefly (#164177) 2025-10-02 20:57:41 +00:00
_subclasses Export should use aot_export_joint_with_descriptors (#165931) 2025-10-27 19:33:33 +00:00
_vendor
accelerator Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
amp [AMP][Refactor] Autocast dtype handling to simplify device-specific c… (#165221) 2025-10-28 06:21:29 +00:00
ao Fix existing pyrefly errors on main (#166312) 2025-10-27 19:03:06 +00:00
autograd Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
backends Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
compiler Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
contrib
cpu Add initial suppressions for pyrefly (#164177) 2025-10-02 20:57:41 +00:00
csrc [GR v0] AOTI Enablement - Fix GR model AOTI inplace update by skipping empty named (#165970) (#166037) 2025-10-28 01:50:36 +00:00
cuda Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
distributed Enable local tensor mode for another set of DTensor tests (#166105) 2025-10-27 23:58:24 +00:00
distributions Fix pyrelfy ignore syntax in distributions and ao (#166248) 2025-10-26 22:13:48 +00:00
export [export] Update dynamo_graph_capture_for_export to return GraphModule. (#166091) 2025-10-28 04:23:28 +00:00
fft
func
futures [5/N] Apply ruff UP035 rule (#164423) 2025-10-02 07:31:11 +00:00
fx Update gm.print_readable to include Annotation (#165397) 2025-10-28 13:54:38 +00:00
headeronly Remove likely unnecessary _EXPAND trick for non-windows in HIDDEN_NAMESPACE_BEGIN (#166203) 2025-10-25 04:44:07 +00:00
jit [dynamo, 3.14] fix misc. bugs to get most dynamo unittests passing locally in 3.14 (#164631) 2025-10-28 03:24:22 +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 flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +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 flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +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 docs: fix typos (#164879) 2025-10-28 12:00:36 +00:00
numa Add initial suppressions for pyrefly (#164177) 2025-10-02 20:57:41 +00:00
onnx Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +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 [EZ][MPS] Improve distribution error checking (#166425) 2025-10-28 18:42:12 +00:00
utils [torch/utils][Code Clean] Clean asserts in torch/utils/*.py (#165410) 2025-10-27 19:48:55 +00:00
xpu Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
__config__.py
__future__.py
__init__.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +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 flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
_linalg_utils.py
_lobpcg.py Add initial suppressions for pyrefly (#164177) 2025-10-02 20:57:41 +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 ignores 1/n (#166239) 2025-10-26 00:44:10 +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 Enable all PIE rules on ruff (#165814) 2025-10-18 07:36:18 +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 Clean up unused Pyrefly suppressions (#166178) 2025-10-25 05:32:21 +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 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