Fix inlining id functions

hercules_opt/src/inline.rs

@@ -210,7 +210,11 @@ fn inline_func(
             }
 
             // Finally, delete the call node.
-            edit = edit.replace_all_uses(id, old_id_to_new_id(called_return_data))?;
+            if let Node::Parameter { index } = called_func.nodes[called_return_data.idx()] {
+                edit = edit.replace_all_uses(id, args[index])?;
+            } else {
+                edit = edit.replace_all_uses(id, old_id_to_new_id(called_return_data))?;
+            }
             edit = edit.delete_node(control)?;
             edit = edit.delete_node(id)?;
 

juno_samples/fork_join_tests/src/gpu.sch

@@ -52,6 +52,7 @@ slf(auto.test2);
 infer-schedules(auto.test2);
 fork-interchange[0, 1](auto.test2);
 
+inline(test6);
 fork-tile[32, 0, false, true](test6@loop);
 let out = fork-split(test6@loop);
 let out = auto-outline(test6);