fix

Author: yf225

helion/_compiler/device_ir.py

@@ -1166,8 +1166,9 @@ def visit_Subscript(self, node: ast.Subscript) -> object:
         assert isinstance(value, ExtendedAST)
         type_info = value._type_info
         if isinstance(type_info, SequenceType):
-            if isinstance(node.slice, ast.Constant):
-                return self.visit(value)[self.visit(node.slice)]  # pyright: ignore[reportIndexIssue]
+            index_value = self.visit(node.slice)
+            if isinstance(index_value, int):
+                return self.visit(value)[index_value]  # pyright: ignore[reportIndexIssue]
             raise exc.InvalidSequenceSubscription(node.slice)
         if isinstance(type_info, StackTensorType):
             return hl.load(self.visit(value), self._subscript_slice_proxy(node.slice))  # pyright: ignore[reportArgumentType]

helion/_compiler/type_propagation.py

@@ -1273,6 +1273,17 @@ def populate_symbol_origins(self, origin: Origin) -> None:
             subtype.populate_symbol_origins(GetItemOrigin(origin, i))
 
     def propagate_getitem(self, key: TypeInfo, origin: Origin) -> TypeInfo:
+        # Tuple indexing with non-literal indices (e.g., from hl.static_range)
+        if self.python_type is tuple and isinstance(key, SymIntType):
+            if not self.element_types:
+                raise exc.TypeInferenceError("Cannot index empty tuple")
+            first_type = self.element_types[0]
+            if not all(type(e) is type(first_type) for e in self.element_types[1:]):
+                raise exc.TypeInferenceError(
+                    "Tuple indexing with non-literal index requires all elements to have the same type"
+                )
+            return first_type
+
         return super().propagate_getitem(key, origin)
 
     def merge(self, other: TypeInfo, var_name: str | None = None) -> TypeInfo: