Fix type for keep_inference_mutation flag (#114482)

tugsbayasgalan · pytorchmergebot · commit dad3cc4d0267 · 2023-11-24T00:04:31.000Z
Pull Request resolved: #114482 Approved by: https://github.com/Skylion007 ghstack dependencies: #114421, #114479, #114481
diff --git a/torch/_functorch/aot_autograd.py b/torch/_functorch/aot_autograd.py
@@ -564,7 +564,7 @@ class ViewAndMutationMeta:
     num_intermediate_bases: int
 
     # For inference only: instructs us to keep data-only input mutations directly in the graph
-    keep_input_mutations: int
+    keep_input_mutations: bool
 
     # length = (# inputs w data mutations) + (# user outputs that are non_aliasing tensors)
     #        + (# intermediate bases)
