Skip to content

Commit f7e9728

Browse files
yf225tianrengao
authored andcommitted
[Fix upcoming CI error] Set current node in inductor lowering (#1052)
1 parent 5d108d5 commit f7e9728

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

helion/_compiler/inductor_lowering.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,7 @@ def convert_arg(arg: Node) -> TensorBox:
177177
prior_buffers = len(graph_lowering.buffers)
178178
input_names: list[str] = []
179179
with inductor_config.patch(INDUCTOR_PATCH):
180-
with node.meta["location"]:
180+
with node.meta["location"], graph_lowering.set_current_node(node):
181181
try:
182182
result = graph_lowering.call_function(
183183
node.target, # pyright: ignore[reportArgumentType]

0 commit comments

Comments
 (0)