lintrunner

pytorch · Sep 7, 2023 · 9077584 · 9077584
1 parent 76d2a97
commit 9077584
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/torch/_inductor/codegen/wrapper.py b/torch/_inductor/codegen/wrapper.py
@@ -1021,12 +1021,14 @@ def write_wrapper_decl(self):
             for idx, constants_key in enumerate(V.graph.constants.keys()):
                 if V.graph.aot_mode:
                     self.prefix.writeline(
-                        f"""at::Tensor {constants_key} = constants_->at("{constants_key}");"""
+                        f"""auto {constants_key} = constants_->at("{constants_key}");"""
                     )
                 else:
                     # Append constants as inputs to the graph
                     constants_idx = inputs_len + idx
-                    self.prefix.writeline(f"auto {constants_key} = args[{constants_idx}];")
+                    self.prefix.writeline(
+                        f"auto {constants_key} = args[{constants_idx}];"
+                    )
 
             self.codegen_inputs(self.prefix, V.graph.graph_inputs)