frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Revert "Enable all SIM rules except disabled ones (#164645)"

Browse Source 
This reverts commit 321e6026925f6b6e8a36e3a8b7c0295cd7541911.

Reverted https://github.com/pytorch/pytorch/pull/164645 on behalf of https://github.com/izaitsevfb due to causes lint failures ([comment](https://github.com/pytorch/pytorch/pull/164645#issuecomment-3369274351))
This commit is contained in:
PyTorch MergeBot
2025-10-05 19:32:21 +00:00
parent 321e602692
commit 5d7360bb03
97 changed files with 255 additions and 182 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torchgen/gen_vmap_plumbing.py
View File

@ -150,7 +150,7 @@ def gen_vmap_inplace_plumbing(native_function: NativeFunction) -> str | None:
assert schema.kind() == SchemaKind.inplace
if not is_mutated_arg(schema.arguments.flat_all[0]):
return None
if len([arg for arg in schema.arguments.flat_all if is_mutated_arg(arg)]) != 1:
if not len([arg for arg in schema.arguments.flat_all if is_mutated_arg(arg)]) == 1:
return None
# Only support cases where all returns are Tensors or vector<Tensor>