saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Refactor functions in functorch for functional (#141808)

Browse source 
As the title stated
Pull Request resolved: https://github.com/pytorch/pytorch/pull/141808
Approved by: https://github.com/Skylion007
This commit is contained in:
FFFrog 2024-11-30 09:39:31 +08:00 committed by PyTorch MergeBot
parent 0f3f801fc2
commit 8a81f7a4b6
1 changed files with 1 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
torch/csrc/functorch/init.cpp
View file

@ -75,8 +75,7 @@ void _propagate_functional_input_mutation(
// It would probably be more reasonable to check that the two tensors are
// aliased, but we can't do that unless we give BatchedTensorImpl a notion of
// storage.
if (unwrapped.unsafeGetTensorImpl() == wrapped_inner.unsafeGetTensorImpl()) {
} else {
if (unwrapped.unsafeGetTensorImpl() != wrapped_inner.unsafeGetTensorImpl()) {
if (unwrapped.sym_nbytes() != wrapped_inner.sym_nbytes()) {
// Functions might resize zero-sized inputs, which we need to reflect
// ehre.