commit a1edf5f63c62d88230d1f7feb26edb059551ae71
author Edward Z. Yang <ezyang@meta.com> Wed Mar 22 13:37:52 2023 -0400
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Wed Mar 22 23:26:00 2023 +0000
tree 7691385e89cd8d80881ce8e8fac555a062a76926
parent 5425191f574fdc83c02dc576e3d5f58c1c6c8b72

[EASY] Do hook sizes check with SymInt (#97362)

I don't think this matters for any uses right now, but I found
it during an audit; might as well fix it.

Signed-off-by: Edward Z. Yang <ezyang@meta.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/97362
Approved by: https://github.com/wconstab

torch/csrc/autograd/custom_function.cpp

diff --git a/torch/csrc/autograd/custom_function.cpp b/torch/csrc/autograd/custom_function.cpp
index 527a87a..e418feb 100644
--- a/torch/csrc/autograd/custom_function.cpp
+++ b/torch/csrc/autograd/custom_function.cpp
@@ -474,7 +474,7 @@
     throw std::runtime_error(ss.str());
   }
 
-  if (original.sizes().vec() != result.sizes().vec()) {
+  if (original.sym_sizes().vec() != result.sym_sizes().vec()) {
     std::stringstream ss;
     ss << "hook '" << hook_name << "' has changed the size of value";
     throw std::runtime_error(ss.str());