pytorch
diff --git a/‎torch/_inductor/codegen/cpp_template_kernel.py
Copy file name to clipboard
+1-1Lines changed: 1 addition & 1 deletion b/‎torch/_inductor/codegen/cpp_template_kernel.py
Copy file name to clipboard
+1-1Lines changed: 1 addition & 1 deletion
diff --git a/‎torch/_inductor/dependencies.py
Copy file name to clipboard
+9-7Lines changed: 9 additions & 7 deletions b/‎torch/_inductor/dependencies.py
Copy file name to clipboard
+9-7Lines changed: 9 additions & 7 deletions
@@ -220,7 +220,7 @@ def store_pointwise_nodes(
         if not reindexers:
             reindexers = [None] * len(nodes)
         assert len(offsets) == len(var_sizes[0])
-        output_index = dst.get_layout().make_indexer()(var_ranges.keys())
+        output_index = dst.get_layout().make_indexer()([*var_ranges.keys()])
         kernel_group = KernelGroup()
         kernel_group.args = self.args
         cpp_kernel_proxy = CppKernelProxy(kernel_group)
 
@@ -641,7 +641,7 @@ def extract_read_writes(
 
 
 def extract_input_node_reduction_ranges(
-    input_node: "torch._inductor.ir.TensorBox",
+    input_node: "torch._inductor.ir.IRNode",
 ) -> Tuple[Optional[List[sympy.Expr]], Optional[List[sympy.Expr]]]:
     """
     Returns the size and reduction size of all inputs, if the sizes and reduction_sizes (if exist) are all the same.
@@ -655,7 +655,7 @@ def extract_input_node_reduction_ranges(
     size: Optional[List[sympy.Expr]]
     reduction_size: Optional[List[sympy.Expr]]
 
-    if isinstance(input_node.data, ComputedBuffer):
+    if isinstance(input_node.get_defining_op(), ComputedBuffer):
         # Input node has already been realized. Return its size and reduction_size.
         size = [*input_node.get_size()]
         reduction_size = [*input_node.get_reduction_size()]
@@ -672,8 +672,8 @@ def extract_input_node_reduction_ranges(
     # The current method still uses reduction ranges from the dependent realized node, which is not ideal.
     # Is there a way to check whether there are permutations inbetween?
     reads = input_node.get_reads()
-    reduction_size = None
-    size = None
+    reduction_size: Optional[List[sympy.Expr]] = None
+    size: Optional[List[sympy.Expr]] = None
     while reduction_size is None and len(reads) > 0:
         seen: OrderedSet[str] = OrderedSet()
         new_reads: List[Dep] = []
@@ -692,9 +692,11 @@ def extract_input_node_reduction_ranges(
 
             if isinstance(op, ComputedBuffer) and len(op.get_reduction_size()) > 0:
                 if reduction_size is None:
-                    reduction_size = op.get_reduction_size()
-                    size = op.get_size()
-                elif reduction_size != op.get_reduction_size() or size != op.get_size():
+                    reduction_size = [*op.get_reduction_size()]
+                    size = [*op.get_size()]
+                elif reduction_size != [*op.get_reduction_size()] or size != [
+                    *op.get_size()
+                ]:
                     return (None, None)
             else:
                 new_reads.extend(op.get_reads())