Use wrapper instead of V.graph.wrapper_code (#109883)
Signed-off-by: Edward Z. Yang <ezyang@meta.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/109883
Approved by: https://github.com/voznesenskym, https://github.com/jansel
diff --git a/torch/_inductor/ir.py b/torch/_inductor/ir.py
index 807e5ee..b9d9450 100644
--- a/torch/_inductor/ir.py
+++ b/torch/_inductor/ir.py
@@ -3911,11 +3911,11 @@
return basename
def codegen(self, wrapper):
- line = V.graph.wrapper_code.declare
+ line = wrapper.declare
line += f"{self.get_name()} = {self.codegen_list_tuple_access(self.inputs[0].get_name(), self.indices)}"
- line += V.graph.wrapper_code.ending
- V.graph.wrapper_code.writeline(line)
- self.codegen_size_asserts(V.graph.wrapper_code)
+ line += wrapper.ending
+ wrapper.writeline(line)
+ self.codegen_size_asserts(wrapper)
def __init__(self, layout, input, indices: List[Tuple[Any, ...]]):
super().__init__(None, layout, [input], ())
