/*
 * Copyright (c) 2012, 2015, Oracle and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */


package org.graalvm.compiler.phases.common.inlining;

import static jdk.vm.ci.meta.DeoptimizationAction.InvalidateReprofile;
import static jdk.vm.ci.meta.DeoptimizationReason.NullCheckException;
import static org.graalvm.compiler.core.common.GraalOptions.HotSpotPrintInlining;

import java.lang.reflect.Constructor;
import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.function.Consumer;

import jdk.internal.vm.compiler.collections.EconomicMap;
import jdk.internal.vm.compiler.collections.EconomicSet;
import jdk.internal.vm.compiler.collections.Equivalence;
import jdk.internal.vm.compiler.collections.UnmodifiableEconomicMap;
import jdk.internal.vm.compiler.collections.UnmodifiableMapCursor;
import org.graalvm.compiler.api.replacements.MethodSubstitution;
import org.graalvm.compiler.api.replacements.Snippet;
import org.graalvm.compiler.core.common.GraalOptions;
import org.graalvm.compiler.core.common.type.Stamp;
import org.graalvm.compiler.core.common.type.StampFactory;
import org.graalvm.compiler.core.common.type.TypeReference;
import org.graalvm.compiler.core.common.util.Util;
import org.graalvm.compiler.debug.DebugCloseable;
import org.graalvm.compiler.debug.DebugContext;
import org.graalvm.compiler.debug.GraalError;
import org.graalvm.compiler.graph.GraalGraphError;
import org.graalvm.compiler.graph.Graph.DuplicationReplacement;
import org.graalvm.compiler.graph.Graph.Mark;
import org.graalvm.compiler.graph.Graph.NodeEventScope;
import org.graalvm.compiler.graph.Node;
import org.graalvm.compiler.graph.NodeInputList;
import org.graalvm.compiler.graph.NodeMap;
import org.graalvm.compiler.graph.NodeSourcePosition;
import org.graalvm.compiler.graph.NodeWorkList;
import org.graalvm.compiler.nodeinfo.Verbosity;
import org.graalvm.compiler.nodes.AbstractBeginNode;
import org.graalvm.compiler.nodes.AbstractEndNode;
import org.graalvm.compiler.nodes.AbstractMergeNode;
import org.graalvm.compiler.nodes.BeginNode;
import org.graalvm.compiler.nodes.CallTargetNode;
import org.graalvm.compiler.nodes.CallTargetNode.InvokeKind;
import org.graalvm.compiler.nodes.DeoptimizeNode;
import org.graalvm.compiler.nodes.DeoptimizingGuard;
import org.graalvm.compiler.nodes.EndNode;
import org.graalvm.compiler.nodes.FixedGuardNode;
import org.graalvm.compiler.nodes.FixedNode;
import org.graalvm.compiler.nodes.FixedWithNextNode;
import org.graalvm.compiler.nodes.FrameState;
import org.graalvm.compiler.nodes.InliningLog;
import org.graalvm.compiler.nodes.Invoke;
import org.graalvm.compiler.nodes.InvokeNode;
import org.graalvm.compiler.nodes.InvokeWithExceptionNode;
import org.graalvm.compiler.nodes.KillingBeginNode;
import org.graalvm.compiler.nodes.LogicNode;
import org.graalvm.compiler.nodes.MergeNode;
import org.graalvm.compiler.nodes.NodeView;
import org.graalvm.compiler.nodes.ParameterNode;
import org.graalvm.compiler.nodes.PhiNode;
import org.graalvm.compiler.nodes.PiNode;
import org.graalvm.compiler.nodes.ReturnNode;
import org.graalvm.compiler.nodes.StartNode;
import org.graalvm.compiler.nodes.StateSplit;
import org.graalvm.compiler.nodes.StructuredGraph;
import org.graalvm.compiler.nodes.StructuredGraph.GuardsStage;
import org.graalvm.compiler.nodes.UnwindNode;
import org.graalvm.compiler.nodes.ValueNode;
import org.graalvm.compiler.nodes.calc.IsNullNode;
import org.graalvm.compiler.nodes.extended.ForeignCallNode;
import org.graalvm.compiler.nodes.extended.GuardingNode;
import org.graalvm.compiler.nodes.java.ExceptionObjectNode;
import org.graalvm.compiler.nodes.java.MethodCallTargetNode;
import org.graalvm.compiler.nodes.java.MonitorExitNode;
import org.graalvm.compiler.nodes.java.MonitorIdNode;
import org.graalvm.compiler.nodes.spi.Replacements;
import org.graalvm.compiler.nodes.type.StampTool;
import org.graalvm.compiler.nodes.util.GraphUtil;
import org.graalvm.compiler.phases.common.inlining.info.InlineInfo;
import org.graalvm.compiler.phases.common.util.HashSetNodeEventListener;
import org.graalvm.compiler.phases.util.ValueMergeUtil;

import jdk.vm.ci.code.BytecodeFrame;
import jdk.vm.ci.meta.Assumptions;
import jdk.vm.ci.meta.DeoptimizationAction;
import jdk.vm.ci.meta.DeoptimizationReason;
import jdk.vm.ci.meta.JavaKind;
import jdk.vm.ci.meta.ResolvedJavaMethod;
import jdk.vm.ci.meta.ResolvedJavaType;

public class InliningUtil extends ValueMergeUtil {

    private static final String inliningDecisionsScopeString = "InliningDecisions";

    
Print a HotSpot-style inlining message to the console.
/** * Print a HotSpot-style inlining message to the console. */
private static void printInlining(final InlineInfo info, final int inliningDepth, final boolean success, final String msg, final Object... args) { printInlining(info.methodAt(0), info.invoke(), inliningDepth, success, msg, args); }
See Also:
  • printInlining
/** * @see #printInlining */
private static void printInlining(final ResolvedJavaMethod method, final Invoke invoke, final int inliningDepth, final boolean success, final String msg, final Object... args) { if (HotSpotPrintInlining.getValue(invoke.asNode().getOptions())) { Util.printInlining(method, invoke.bci(), inliningDepth, success, msg, args); } }
Trace a decision to inline a method. This prints a HotSpot-style inlining message to the console, and it also logs the decision to the logging stream. Phases that perform inlining should use this method to trace the inlining decisions, and use the traceNotInlinedMethod methods only for debugging purposes.
/** * Trace a decision to inline a method. * * This prints a HotSpot-style inlining message to the console, and it also logs the decision to * the logging stream. * * Phases that perform inlining should use this method to trace the inlining decisions, and use * the {@link #traceNotInlinedMethod} methods only for debugging purposes. */
public static void traceInlinedMethod(InlineInfo info, int inliningDepth, boolean allowLogging, String msg, Object... args) { traceMethod(info, inliningDepth, allowLogging, true, msg, args); }
Trace a decision to inline a method. This prints a HotSpot-style inlining message to the console, and it also logs the decision to the logging stream. Phases that perform inlining should use this method to trace the inlining decisions, and use the traceNotInlinedMethod methods only for debugging purposes.
/** * Trace a decision to inline a method. * * This prints a HotSpot-style inlining message to the console, and it also logs the decision to * the logging stream. * * Phases that perform inlining should use this method to trace the inlining decisions, and use * the {@link #traceNotInlinedMethod} methods only for debugging purposes. */
public static void traceInlinedMethod(Invoke invoke, int inliningDepth, boolean allowLogging, ResolvedJavaMethod method, String msg, Object... args) { traceMethod(invoke, inliningDepth, allowLogging, true, method, msg, args); }
Trace a decision to not inline a method. This prints a HotSpot-style inlining message to the console, and it also logs the decision to the logging stream. Phases that perform inlining should use this method to trace the inlining decisions, and use the traceNotInlinedMethod methods only for debugging purposes.
/** * Trace a decision to not inline a method. * * This prints a HotSpot-style inlining message to the console, and it also logs the decision to * the logging stream. * * Phases that perform inlining should use this method to trace the inlining decisions, and use * the {@link #traceNotInlinedMethod} methods only for debugging purposes. */
public static void traceNotInlinedMethod(InlineInfo info, int inliningDepth, String msg, Object... args) { traceMethod(info, inliningDepth, true, false, msg, args); }
Trace a decision about not inlining a method. This prints a HotSpot-style inlining message to the console, and it also logs the decision to the logging stream. Phases that perform inlining should use this method to trace the inlining decisions, and use the traceNotInlinedMethod methods only for debugging purposes.
/** * Trace a decision about not inlining a method. * * This prints a HotSpot-style inlining message to the console, and it also logs the decision to * the logging stream. * * Phases that perform inlining should use this method to trace the inlining decisions, and use * the {@link #traceNotInlinedMethod} methods only for debugging purposes. */
public static void traceNotInlinedMethod(Invoke invoke, int inliningDepth, ResolvedJavaMethod method, String msg, Object... args) { traceMethod(invoke, inliningDepth, true, false, method, msg, args); } private static void traceMethod(Invoke invoke, int inliningDepth, boolean allowLogging, boolean success, ResolvedJavaMethod method, String msg, Object... args) { if (allowLogging) { DebugContext debug = invoke.asNode().getDebug(); printInlining(method, invoke, inliningDepth, success, msg, args); if (shouldLogMethod(debug)) { String methodString = methodName(method, invoke); logMethod(debug, methodString, success, msg, args); } } } private static void traceMethod(InlineInfo info, int inliningDepth, boolean allowLogging, boolean success, String msg, final Object... args) { if (allowLogging) { printInlining(info, inliningDepth, success, msg, args); DebugContext debug = info.graph().getDebug(); if (shouldLogMethod(debug)) { logMethod(debug, methodName(info), success, msg, args); } } }
Output a generic inlining decision to the logging stream (e.g. inlining termination condition). Used for debugging purposes.
/** * Output a generic inlining decision to the logging stream (e.g. inlining termination * condition). * * Used for debugging purposes. */
public static void logInliningDecision(DebugContext debug, final String msg, final Object... args) { logInlining(debug, msg, args); }
Output a decision about not inlining a method to the logging stream, for debugging purposes.
/** * Output a decision about not inlining a method to the logging stream, for debugging purposes. */
public static void logNotInlinedMethod(Invoke invoke, String msg) { DebugContext debug = invoke.asNode().getDebug(); if (shouldLogMethod(debug)) { String methodString = invoke.toString(); if (invoke.callTarget() == null) { methodString += " callTarget=null"; } else { String targetName = invoke.callTarget().targetName(); if (!methodString.endsWith(targetName)) { methodString += " " + targetName; } } logMethod(debug, methodString, false, msg, new Object[0]); } } private static void logMethod(DebugContext debug, final String methodString, final boolean success, final String msg, final Object... args) { String inliningMsg = "inlining " + methodString + ": " + msg; if (!success) { inliningMsg = "not " + inliningMsg; } logInlining(debug, inliningMsg, args); } @SuppressWarnings("try") private static void logInlining(DebugContext debug, final String msg, final Object... args) { try (DebugContext.Scope s = debug.scope(inliningDecisionsScopeString)) { // Can't use log here since we are varargs if (debug.isLogEnabled()) { debug.logv(msg, args); } } } @SuppressWarnings("try") private static boolean shouldLogMethod(DebugContext debug) { try (DebugContext.Scope s = debug.scope(inliningDecisionsScopeString)) { return debug.isLogEnabled(); } } private static String methodName(ResolvedJavaMethod method, Invoke invoke) { if (invoke != null && invoke.stateAfter() != null) { return methodName(invoke.stateAfter(), invoke.bci()) + ": " + method.format("%H.%n(%p):%r") + " (" + method.getCodeSize() + " bytes)"; } else { return method.format("%H.%n(%p):%r") + " (" + method.getCodeSize() + " bytes)"; } } private static String methodName(InlineInfo info) { if (info == null) { return "null"; } else if (info.invoke() != null && info.invoke().stateAfter() != null) { return methodName(info.invoke().stateAfter(), info.invoke().bci()) + ": " + info.toString(); } else { return info.toString(); } } private static String methodName(FrameState frameState, int bci) { StringBuilder sb = new StringBuilder(); if (frameState.outerFrameState() != null) { sb.append(methodName(frameState.outerFrameState(), frameState.outerFrameState().bci)); sb.append("->"); } ResolvedJavaMethod method = frameState.getMethod(); sb.append(method != null ? method.format("%h.%n") : "?"); sb.append("@").append(bci); return sb.toString(); } public static void replaceInvokeCallTarget(Invoke invoke, StructuredGraph graph, InvokeKind invokeKind, ResolvedJavaMethod targetMethod) { MethodCallTargetNode oldCallTarget = (MethodCallTargetNode) invoke.callTarget(); MethodCallTargetNode newCallTarget = graph.add(new MethodCallTargetNode(invokeKind, targetMethod, oldCallTarget.arguments().toArray(new ValueNode[0]), oldCallTarget.returnStamp(), oldCallTarget.getProfile())); invoke.asNode().replaceFirstInput(oldCallTarget, newCallTarget); } public static PiNode createAnchoredReceiver(StructuredGraph graph, GuardingNode anchor, ResolvedJavaType commonType, ValueNode receiver, boolean exact) { return createAnchoredReceiver(graph, anchor, receiver, exact ? StampFactory.objectNonNull(TypeReference.createExactTrusted(commonType)) : StampFactory.objectNonNull(TypeReference.createTrusted(graph.getAssumptions(), commonType))); } private static PiNode createAnchoredReceiver(StructuredGraph graph, GuardingNode anchor, ValueNode receiver, Stamp stamp) { // to avoid that floating reads on receiver fields float above the type check return graph.unique(new PiNode(receiver, stamp, (ValueNode) anchor)); }
Returns:null iff the check succeeds, otherwise a (non-null) descriptive message.
/** * @return null iff the check succeeds, otherwise a (non-null) descriptive message. */
public static String checkInvokeConditions(Invoke invoke) { if (invoke.predecessor() == null || !invoke.asNode().isAlive()) { return "the invoke is dead code"; } if (!(invoke.callTarget() instanceof MethodCallTargetNode)) { return "the invoke has already been lowered, or has been created as a low-level node"; } MethodCallTargetNode callTarget = (MethodCallTargetNode) invoke.callTarget(); if (callTarget.targetMethod() == null) { return "target method is null"; } assert invoke.stateAfter() != null : invoke; if (!invoke.useForInlining()) { return "the invoke is marked to be not used for inlining"; } ValueNode receiver = callTarget.receiver(); if (receiver != null && receiver.isConstant() && receiver.isNullConstant()) { return "receiver is null"; } return null; }
Performs an actual inlining, thereby replacing the given invoke with the given inlineGraph.
Params:
  • invoke – the invoke that will be replaced
  • inlineGraph – the graph that the invoke will be replaced with
  • receiverNullCheck – true if a null check needs to be generated for non-static inlinings, false if no such check is required
  • inlineeMethod – the actual method being inlined. Maybe be null for snippets.
/** * Performs an actual inlining, thereby replacing the given invoke with the given * {@code inlineGraph}. * * @param invoke the invoke that will be replaced * @param inlineGraph the graph that the invoke will be replaced with * @param receiverNullCheck true if a null check needs to be generated for non-static inlinings, * false if no such check is required * @param inlineeMethod the actual method being inlined. Maybe be null for snippets. */
@SuppressWarnings("try") public static UnmodifiableEconomicMap<Node, Node> inline(Invoke invoke, StructuredGraph inlineGraph, boolean receiverNullCheck, ResolvedJavaMethod inlineeMethod) { try { return inline(invoke, inlineGraph, receiverNullCheck, inlineeMethod, "reason not specified", "phase not specified"); } catch (GraalError ex) { ex.addContext("inlining into", invoke.asNode().graph().method()); ex.addContext("inlinee", inlineGraph.method()); throw ex; } }
Performs an actual inlining, thereby replacing the given invoke with the given inlineGraph.
Params:
  • invoke – the invoke that will be replaced
  • inlineGraph – the graph that the invoke will be replaced with
  • receiverNullCheck – true if a null check needs to be generated for non-static inlinings, false if no such check is required
  • inlineeMethod – the actual method being inlined. Maybe be null for snippets.
  • reason – the reason for inlining, used in tracing
  • phase – the phase that invoked inlining
/** * Performs an actual inlining, thereby replacing the given invoke with the given * {@code inlineGraph}. * * @param invoke the invoke that will be replaced * @param inlineGraph the graph that the invoke will be replaced with * @param receiverNullCheck true if a null check needs to be generated for non-static inlinings, * false if no such check is required * @param inlineeMethod the actual method being inlined. Maybe be null for snippets. * @param reason the reason for inlining, used in tracing * @param phase the phase that invoked inlining */
@SuppressWarnings("try") public static UnmodifiableEconomicMap<Node, Node> inline(Invoke invoke, StructuredGraph inlineGraph, boolean receiverNullCheck, ResolvedJavaMethod inlineeMethod, String reason, String phase) { FixedNode invokeNode = invoke.asNode(); StructuredGraph graph = invokeNode.graph(); final NodeInputList<ValueNode> parameters = invoke.callTarget().arguments(); assert inlineGraph.getGuardsStage().ordinal() >= graph.getGuardsStage().ordinal(); assert !invokeNode.graph().isAfterFloatingReadPhase() : "inline isn't handled correctly after floating reads phase"; if (receiverNullCheck && !((MethodCallTargetNode) invoke.callTarget()).isStatic()) { nonNullReceiver(invoke); } ArrayList<Node> nodes = new ArrayList<>(inlineGraph.getNodes().count()); ArrayList<ReturnNode> returnNodes = new ArrayList<>(4); ArrayList<Invoke> partialIntrinsicExits = new ArrayList<>(); UnwindNode unwindNode = null; final StartNode entryPointNode = inlineGraph.start(); FixedNode firstCFGNode = entryPointNode.next(); if (firstCFGNode == null) { throw new IllegalStateException("Inlined graph is in invalid state: " + inlineGraph); } for (Node node : inlineGraph.getNodes()) { if (node == entryPointNode || (node == entryPointNode.stateAfter() && node.usages().count() == 1) || node instanceof ParameterNode) { // Do nothing. } else { nodes.add(node); if (node instanceof ReturnNode) { returnNodes.add((ReturnNode) node); } else if (node instanceof Invoke) { Invoke invokeInInlineGraph = (Invoke) node; if (invokeInInlineGraph.bci() == BytecodeFrame.UNKNOWN_BCI) { ResolvedJavaMethod target1 = inlineeMethod; ResolvedJavaMethod target2 = invokeInInlineGraph.callTarget().targetMethod(); assert target1.equals(target2) : String.format("invoke in inlined method expected to be partial intrinsic exit (i.e., call to %s), not a call to %s", target1.format("%H.%n(%p)"), target2.format("%H.%n(%p)")); partialIntrinsicExits.add(invokeInInlineGraph); } } else if (node instanceof UnwindNode) { assert unwindNode == null; unwindNode = (UnwindNode) node; } } } final AbstractBeginNode prevBegin = AbstractBeginNode.prevBegin(invokeNode); DuplicationReplacement localReplacement = new DuplicationReplacement() { @Override public Node replacement(Node node) { if (node instanceof ParameterNode) { return parameters.get(((ParameterNode) node).index()); } else if (node == entryPointNode) { return prevBegin; } return node; } }; assert invokeNode.successors().first() != null : invoke; assert invokeNode.predecessor() != null; Mark mark = graph.getMark(); // Instead, attach the inlining log of the child graph to the current inlining log. EconomicMap<Node, Node> duplicates; try (InliningLog.UpdateScope scope = graph.getInliningLog().openDefaultUpdateScope()) { duplicates = graph.addDuplicates(nodes, inlineGraph, inlineGraph.getNodeCount(), localReplacement); if (scope != null) { graph.getInliningLog().addDecision(invoke, true, phase, duplicates, inlineGraph.getInliningLog(), reason); } } FrameState stateAfter = invoke.stateAfter(); assert stateAfter == null || stateAfter.isAlive(); FrameState stateAtExceptionEdge = null; if (invoke instanceof InvokeWithExceptionNode) { InvokeWithExceptionNode invokeWithException = ((InvokeWithExceptionNode) invoke); if (unwindNode != null) { ExceptionObjectNode obj = (ExceptionObjectNode) invokeWithException.exceptionEdge(); stateAtExceptionEdge = obj.stateAfter(); } } updateSourcePositions(invoke, inlineGraph, duplicates, !Objects.equals(inlineGraph.method(), inlineeMethod), mark); if (stateAfter != null) { processFrameStates(invoke, inlineGraph, duplicates, stateAtExceptionEdge, returnNodes.size() > 1); int callerLockDepth = stateAfter.nestedLockDepth(); if (callerLockDepth != 0) { for (MonitorIdNode original : inlineGraph.getNodes(MonitorIdNode.TYPE)) { MonitorIdNode monitor = (MonitorIdNode) duplicates.get(original); processMonitorId(invoke.stateAfter(), monitor); } } } else { assert checkContainsOnlyInvalidOrAfterFrameState(duplicates); } firstCFGNode = (FixedNode) duplicates.get(firstCFGNode); for (int i = 0; i < returnNodes.size(); i++) { returnNodes.set(i, (ReturnNode) duplicates.get(returnNodes.get(i))); } for (Invoke exit : partialIntrinsicExits) { // A partial intrinsic exit must be replaced with a call to // the intrinsified method. Invoke dup = (Invoke) duplicates.get(exit.asNode()); if (dup instanceof InvokeNode) { InvokeNode repl = graph.add(new InvokeNode(invoke.callTarget(), invoke.bci())); dup.intrinsify(repl.asNode()); } else { ((InvokeWithExceptionNode) dup).replaceWithNewBci(invoke.bci()); } } if (unwindNode != null) { unwindNode = (UnwindNode) duplicates.get(unwindNode); } finishInlining(invoke, graph, firstCFGNode, returnNodes, unwindNode, inlineGraph.getAssumptions(), inlineGraph); GraphUtil.killCFG(invokeNode); return duplicates; }
Inline inlineGraph into the current replacing the node Invoke and return the set of nodes which should be canonicalized. The set should only contain nodes which modified by the inlining since the current graph and inlineGraph are expected to already be canonical.
Params:
  • invoke –
  • inlineGraph –
  • receiverNullCheck –
  • inlineeMethod –
Returns:the set of nodes to canonicalize
/** * Inline {@code inlineGraph} into the current replacing the node {@code Invoke} and return the * set of nodes which should be canonicalized. The set should only contain nodes which modified * by the inlining since the current graph and {@code inlineGraph} are expected to already be * canonical. * * @param invoke * @param inlineGraph * @param receiverNullCheck * @param inlineeMethod * @return the set of nodes to canonicalize */
@SuppressWarnings("try") public static EconomicSet<Node> inlineForCanonicalization(Invoke invoke, StructuredGraph inlineGraph, boolean receiverNullCheck, ResolvedJavaMethod inlineeMethod, String reason, String phase) { return inlineForCanonicalization(invoke, inlineGraph, receiverNullCheck, inlineeMethod, null, reason, phase); } @SuppressWarnings("try") public static EconomicSet<Node> inlineForCanonicalization(Invoke invoke, StructuredGraph inlineGraph, boolean receiverNullCheck, ResolvedJavaMethod inlineeMethod, Consumer<UnmodifiableEconomicMap<Node, Node>> duplicatesConsumer, String reason, String phase) { HashSetNodeEventListener listener = new HashSetNodeEventListener(); /* * This code relies on the fact that Graph.addDuplicates doesn't trigger the * NodeEventListener to track only nodes which were modified into the process of inlining * the graph into the current graph. */ try (NodeEventScope nes = invoke.asNode().graph().trackNodeEvents(listener)) { UnmodifiableEconomicMap<Node, Node> duplicates = InliningUtil.inline(invoke, inlineGraph, receiverNullCheck, inlineeMethod, reason, phase); if (duplicatesConsumer != null) { duplicatesConsumer.accept(duplicates); } } return listener.getNodes(); } @SuppressWarnings("try") private static ValueNode finishInlining(Invoke invoke, StructuredGraph graph, FixedNode firstNode, List<ReturnNode> returnNodes, UnwindNode unwindNode, Assumptions inlinedAssumptions, StructuredGraph inlineGraph) { FixedNode invokeNode = invoke.asNode(); FrameState stateAfter = invoke.stateAfter(); assert stateAfter == null || stateAfter.isAlive(); invokeNode.replaceAtPredecessor(firstNode); if (invoke instanceof InvokeWithExceptionNode) { InvokeWithExceptionNode invokeWithException = ((InvokeWithExceptionNode) invoke); if (unwindNode != null && unwindNode.isAlive()) { assert unwindNode.predecessor() != null; assert invokeWithException.exceptionEdge().successors().count() == 1; ExceptionObjectNode obj = (ExceptionObjectNode) invokeWithException.exceptionEdge(); obj.replaceAtUsages(unwindNode.exception()); Node n = obj.next(); obj.setNext(null); unwindNode.replaceAndDelete(n); obj.replaceAtPredecessor(null); obj.safeDelete(); } else { invokeWithException.killExceptionEdge(); } // get rid of memory kill AbstractBeginNode begin = invokeWithException.next(); if (begin instanceof KillingBeginNode) { try (DebugCloseable position = begin.withNodeSourcePosition()) { AbstractBeginNode newBegin = new BeginNode(); graph.addAfterFixed(begin, graph.add(newBegin)); begin.replaceAtUsages(newBegin); graph.removeFixed(begin); } } } else { if (unwindNode != null && unwindNode.isAlive()) { try (DebugCloseable position = unwindNode.withNodeSourcePosition()) { DeoptimizeNode deoptimizeNode = addDeoptimizeNode(graph, DeoptimizationAction.InvalidateRecompile, DeoptimizationReason.NotCompiledExceptionHandler); unwindNode.replaceAndDelete(deoptimizeNode); } } } ValueNode returnValue; if (!returnNodes.isEmpty()) { FixedNode n = invoke.next(); invoke.setNext(null); if (returnNodes.size() == 1) { ReturnNode returnNode = returnNodes.get(0); returnValue = returnNode.result(); invokeNode.replaceAtUsages(returnValue); returnNode.replaceAndDelete(n); } else { MergeNode merge = graph.add(new MergeNode()); merge.setStateAfter(stateAfter); returnValue = mergeReturns(merge, returnNodes); invokeNode.replaceAtUsages(returnValue); if (merge.isPhiAtMerge(returnValue)) { fixFrameStates(graph, merge, (PhiNode) returnValue); } merge.setNext(n); } } else { returnValue = null; invokeNode.replaceAtUsages(null); GraphUtil.killCFG(invoke.next()); } // Copy assumptions from inlinee to caller Assumptions assumptions = graph.getAssumptions(); if (assumptions != null) { if (inlinedAssumptions != null) { assumptions.record(inlinedAssumptions); } } else { assert inlinedAssumptions == null : String.format("cannot inline graph (%s) which makes assumptions into a graph (%s) that doesn't", inlineGraph, graph); } // Copy inlined methods from inlinee to caller graph.updateMethods(inlineGraph); // Update the set of accessed fields if (GraalOptions.GeneratePIC.getValue(graph.getOptions())) { graph.updateFields(inlineGraph); } if (inlineGraph.hasUnsafeAccess()) { graph.markUnsafeAccess(); } assert inlineGraph.getSpeculationLog() == null || inlineGraph.getSpeculationLog() == graph.getSpeculationLog() : "Only the root graph should have a speculation log"; return returnValue; } private static void fixFrameStates(StructuredGraph graph, MergeNode originalMerge, PhiNode returnPhi) { // It is possible that some of the frame states that came from AFTER_BCI reference a Phi // node that was created to merge multiple returns. This can create cycles // (see GR-3949 and GR-3957). // To detect this, we follow the control paths starting from the merge node, // split the Phi node inputs at merges and assign the proper input to each frame state. NodeMap<Node> seen = new NodeMap<>(graph); ArrayDeque<Node> workList = new ArrayDeque<>(); ArrayDeque<ValueNode> valueList = new ArrayDeque<>(); workList.push(originalMerge); valueList.push(returnPhi); while (!workList.isEmpty()) { Node current = workList.pop(); ValueNode currentValue = valueList.pop(); if (seen.containsKey(current)) { continue; } seen.put(current, current); if (current instanceof StateSplit && current != originalMerge) { StateSplit stateSplit = (StateSplit) current; FrameState state = stateSplit.stateAfter(); if (state != null && state.values().contains(returnPhi)) { int index = 0; FrameState duplicate = state.duplicate(); for (ValueNode value : state.values()) { if (value == returnPhi) { duplicate.values().set(index, currentValue); } index++; } stateSplit.setStateAfter(duplicate); GraphUtil.tryKillUnused(state); } } if (current instanceof AbstractMergeNode) { AbstractMergeNode currentMerge = (AbstractMergeNode) current; for (EndNode pred : currentMerge.cfgPredecessors()) { ValueNode newValue = currentValue; if (currentMerge.isPhiAtMerge(currentValue)) { PhiNode currentPhi = (PhiNode) currentValue; newValue = currentPhi.valueAt(pred); } workList.push(pred); valueList.push(newValue); } } else if (current.predecessor() != null) { workList.push(current.predecessor()); valueList.push(currentValue); } } } @SuppressWarnings("try") private static void updateSourcePositions(Invoke invoke, StructuredGraph inlineGraph, UnmodifiableEconomicMap<Node, Node> duplicates, boolean isSub, Mark mark) { FixedNode invokeNode = invoke.asNode(); boolean isSubstitution = isSub || inlineGraph.method().getAnnotation(MethodSubstitution.class) != null || inlineGraph.method().getAnnotation(Snippet.class) != null; StructuredGraph invokeGraph = invokeNode.graph(); assert !invokeGraph.trackNodeSourcePosition() || inlineGraph.trackNodeSourcePosition() || isSubstitution : String.format("trackNodeSourcePosition mismatch %s %s != %s %s", invokeGraph, invokeGraph.trackNodeSourcePosition(), inlineGraph, inlineGraph.trackNodeSourcePosition()); if (invokeGraph.trackNodeSourcePosition() && invoke.stateAfter() != null) { final NodeSourcePosition invokePos = invoke.asNode().getNodeSourcePosition(); updateSourcePosition(invokeGraph, duplicates, mark, invokePos, isSubstitution); } } public static void updateSourcePosition(StructuredGraph invokeGraph, UnmodifiableEconomicMap<Node, Node> duplicates, Mark mark, NodeSourcePosition invokePos, boolean isSubstitution) { /* * Not every duplicate node is newly created, so only update the position of the newly * created nodes. */ EconomicSet<Node> newNodes = EconomicSet.create(Equivalence.DEFAULT); newNodes.addAll(invokeGraph.getNewNodes(mark)); EconomicMap<NodeSourcePosition, NodeSourcePosition> posMap = EconomicMap.create(Equivalence.DEFAULT); UnmodifiableMapCursor<Node, Node> cursor = duplicates.getEntries(); ResolvedJavaMethod inlineeRoot = null; while (cursor.advance()) { Node value = cursor.getValue(); if (!newNodes.contains(value)) { continue; } if (isSubstitution && invokePos == null) { // There's no caller information so the source position for this node will be // invalid, so it should be cleared. value.clearNodeSourcePosition(); } else { NodeSourcePosition pos = cursor.getKey().getNodeSourcePosition(); if (pos != null) { if (inlineeRoot == null) { assert (inlineeRoot = pos.getRootMethod()) != null; } else { assert pos.verifyRootMethod(inlineeRoot); } NodeSourcePosition callerPos = posMap.get(pos); if (callerPos == null) { callerPos = pos.addCaller(invokePos, isSubstitution); posMap.put(pos, callerPos); } value.setNodeSourcePosition(callerPos); if (value instanceof DeoptimizingGuard) { ((DeoptimizingGuard) value).addCallerToNoDeoptSuccessorPosition(callerPos.getCaller()); } } else { if (isSubstitution) { /* * If no other position is provided at least attribute the substituted node * to the original invoke. */ value.setNodeSourcePosition(invokePos); } } } } assert invokeGraph.verifySourcePositions(false); } public static void processMonitorId(FrameState stateAfter, MonitorIdNode monitorIdNode) { if (stateAfter != null) { int callerLockDepth = stateAfter.nestedLockDepth(); monitorIdNode.setLockDepth(monitorIdNode.getLockDepth() + callerLockDepth); } } protected static void processFrameStates(Invoke invoke, StructuredGraph inlineGraph, EconomicMap<Node, Node> duplicates, FrameState stateAtExceptionEdge, boolean alwaysDuplicateStateAfter) { FrameState stateAtReturn = invoke.stateAfter(); FrameState outerFrameState = null; JavaKind invokeReturnKind = invoke.asNode().getStackKind(); EconomicMap<Node, Node> replacements = EconomicMap.create(); for (FrameState original : inlineGraph.getNodes(FrameState.TYPE)) { FrameState frameState = (FrameState) duplicates.get(original); if (frameState != null && frameState.isAlive()) { if (outerFrameState == null) { outerFrameState = stateAtReturn.duplicateModifiedDuringCall(invoke.bci(), invokeReturnKind); } processFrameState(frameState, invoke, replacements, inlineGraph.method(), stateAtExceptionEdge, outerFrameState, alwaysDuplicateStateAfter, invoke.callTarget().targetMethod(), invoke.callTarget().arguments()); } } // If processing the frame states replaced any nodes, update the duplicates map. duplicates.replaceAll((key, value) -> replacements.containsKey(value) ? replacements.get(value) : value); } public static FrameState processFrameState(FrameState frameState, Invoke invoke, EconomicMap<Node, Node> replacements, ResolvedJavaMethod inlinedMethod, FrameState stateAtExceptionEdge, FrameState outerFrameState, boolean alwaysDuplicateStateAfter, ResolvedJavaMethod invokeTargetMethod, List<ValueNode> invokeArgsList) { assert outerFrameState == null || !outerFrameState.isDeleted() : outerFrameState; final FrameState stateAtReturn = invoke.stateAfter(); JavaKind invokeReturnKind = invoke.asNode().getStackKind(); if (frameState.bci == BytecodeFrame.AFTER_BCI) { return handleAfterBciFrameState(frameState, invoke, alwaysDuplicateStateAfter); } else if (stateAtExceptionEdge != null && isStateAfterException(frameState)) { // pop exception object from invoke's stateAfter and replace with this frameState's // exception object (top of stack) FrameState stateAfterException = stateAtExceptionEdge; if (frameState.stackSize() > 0 && stateAtExceptionEdge.stackAt(0) != frameState.stackAt(0)) { stateAfterException = stateAtExceptionEdge.duplicateModified(JavaKind.Object, JavaKind.Object, frameState.stackAt(0)); } frameState.replaceAndDelete(stateAfterException); return stateAfterException; } else if ((frameState.bci == BytecodeFrame.UNWIND_BCI && frameState.graph().getGuardsStage() == GuardsStage.FLOATING_GUARDS) || frameState.bci == BytecodeFrame.AFTER_EXCEPTION_BCI) { /* * This path converts the frame states relevant for exception unwinding to * deoptimization. This is only allowed in configurations when Graal compiles code for * speculative execution (e.g., JIT compilation in HotSpot) but not when compiled code * must be deoptimization free (e.g., AOT compilation for native image generation). * There is currently no global flag in StructuredGraph to distinguish such modes, but * the GuardsStage during inlining indicates the mode in which Graal operates. */ handleMissingAfterExceptionFrameState(frameState, invoke, replacements, alwaysDuplicateStateAfter); return frameState; } else if (frameState.bci == BytecodeFrame.BEFORE_BCI) { // This is an intrinsic. Deoptimizing within an intrinsic // must re-execute the intrinsified invocation assert frameState.outerFrameState() == null; ValueNode[] invokeArgs = invokeArgsList.isEmpty() ? NO_ARGS : invokeArgsList.toArray(new ValueNode[invokeArgsList.size()]); FrameState stateBeforeCall = stateAtReturn.duplicateModifiedBeforeCall(invoke.bci(), invokeReturnKind, invokeTargetMethod.getSignature().toParameterKinds(!invokeTargetMethod.isStatic()), invokeArgs); frameState.replaceAndDelete(stateBeforeCall); return stateBeforeCall; } else { // only handle the outermost frame states if (frameState.outerFrameState() == null) { assert checkInlineeFrameState(invoke, inlinedMethod, frameState); frameState.setOuterFrameState(outerFrameState); } return frameState; } } private static FrameState handleAfterBciFrameState(FrameState frameState, Invoke invoke, boolean alwaysDuplicateStateAfter) { FrameState stateAtReturn = invoke.stateAfter(); JavaKind invokeReturnKind = invoke.asNode().getStackKind(); FrameState stateAfterReturn = stateAtReturn; if (frameState.getCode() == null) { // This is a frame state for a side effect within an intrinsic // that was parsed for post-parse intrinsification for (Node usage : frameState.usages()) { if (usage instanceof ForeignCallNode) { // A foreign call inside an intrinsic needs to have // the BCI of the invoke being intrinsified ForeignCallNode foreign = (ForeignCallNode) usage; foreign.setBci(invoke.bci()); } } } // pop return kind from invoke's stateAfter and replace with this frameState's return // value (top of stack) assert !frameState.rethrowException() : frameState; if (frameState.stackSize() > 0 && (alwaysDuplicateStateAfter || stateAfterReturn.stackAt(0) != frameState.stackAt(0))) { // A non-void return value. stateAfterReturn = stateAtReturn.duplicateModified(invokeReturnKind, invokeReturnKind, frameState.stackAt(0)); } else { // A void return value. stateAfterReturn = stateAtReturn.duplicate(); } assert stateAfterReturn.bci != BytecodeFrame.UNKNOWN_BCI; // Return value does no longer need to be limited by the monitor exit. for (MonitorExitNode n : frameState.usages().filter(MonitorExitNode.class)) { n.clearEscapedReturnValue(); } frameState.replaceAndDelete(stateAfterReturn); return stateAfterReturn; } static boolean checkInlineeFrameState(Invoke invoke, ResolvedJavaMethod inlinedMethod, FrameState frameState) { assert frameState.bci != BytecodeFrame.AFTER_EXCEPTION_BCI : frameState; assert frameState.bci != BytecodeFrame.BEFORE_BCI : frameState; assert frameState.bci != BytecodeFrame.UNKNOWN_BCI : frameState; if (frameState.bci != BytecodeFrame.INVALID_FRAMESTATE_BCI) { ResolvedJavaMethod method = frameState.getMethod(); if (method.equals(inlinedMethod)) { // Normal inlining expects all outermost inlinee frame states to // denote the inlinee method } else if (method.equals(invoke.callTarget().targetMethod())) { // This occurs when an intrinsic calls back to the original // method to handle a slow path. During parsing of such a // partial intrinsic, these calls are given frame states // that exclude the outer frame state denoting a position // in the intrinsic code. assert inlinedMethod.getAnnotation( MethodSubstitution.class) != null : "expected an intrinsic when inlinee frame state matches method of call target but does not match the method of the inlinee graph: " + frameState; } else if (method.getName().equals(inlinedMethod.getName())) { // This can happen for method substitutions. } else { throw new AssertionError(String.format("inlinedMethod=%s frameState.method=%s frameState=%s invoke.method=%s", inlinedMethod, method, frameState, invoke.callTarget().targetMethod())); } } return true; } private static final ValueNode[] NO_ARGS = {}; private static boolean isStateAfterException(FrameState frameState) { return frameState.bci == BytecodeFrame.AFTER_EXCEPTION_BCI || (frameState.bci == BytecodeFrame.UNWIND_BCI && !frameState.getMethod().isSynchronized()); } @SuppressWarnings("try") public static FrameState handleMissingAfterExceptionFrameState(FrameState nonReplaceableFrameState, Invoke invoke, EconomicMap<Node, Node> replacements, boolean alwaysDuplicateStateAfter) { StructuredGraph graph = nonReplaceableFrameState.graph(); NodeWorkList workList = graph.createNodeWorkList(); workList.add(nonReplaceableFrameState); for (Node node : workList) { FrameState fs = (FrameState) node; for (Node usage : fs.usages().snapshot()) { if (!usage.isAlive()) { continue; } if (usage instanceof FrameState) { workList.add(usage); } else { StateSplit stateSplit = (StateSplit) usage; FixedNode fixedStateSplit = stateSplit.asNode(); if (fixedStateSplit instanceof AbstractMergeNode) { AbstractMergeNode merge = (AbstractMergeNode) fixedStateSplit; while (merge.isAlive()) { AbstractEndNode end = merge.forwardEnds().first(); try (DebugCloseable position = end.withNodeSourcePosition()) { DeoptimizeNode deoptimizeNode = addDeoptimizeNode(graph, DeoptimizationAction.InvalidateRecompile, DeoptimizationReason.NotCompiledExceptionHandler); end.replaceAtPredecessor(deoptimizeNode); GraphUtil.killCFG(end); } } } else if (fixedStateSplit instanceof ExceptionObjectNode) { // The target invoke does not have an exception edge. This means that the // bytecode parser made the wrong assumption of making an // InvokeWithExceptionNode for the partial intrinsic exit. We therefore // replace the InvokeWithExceptionNode with a normal // InvokeNode -- the deoptimization occurs when the invoke throws. InvokeWithExceptionNode oldInvoke = (InvokeWithExceptionNode) fixedStateSplit.predecessor(); FrameState oldFrameState = oldInvoke.stateAfter(); InvokeNode newInvoke = oldInvoke.replaceWithInvoke(); newInvoke.setStateAfter(oldFrameState.duplicate()); if (replacements != null) { replacements.put(oldInvoke, newInvoke); } handleAfterBciFrameState(newInvoke.stateAfter(), invoke, alwaysDuplicateStateAfter); } else { try (DebugCloseable position = fixedStateSplit.withNodeSourcePosition()) { FixedNode deoptimizeNode = addDeoptimizeNode(graph, DeoptimizationAction.InvalidateRecompile, DeoptimizationReason.NotCompiledExceptionHandler); if (fixedStateSplit instanceof AbstractBeginNode) { deoptimizeNode = BeginNode.begin(deoptimizeNode); } fixedStateSplit.replaceAtPredecessor(deoptimizeNode); GraphUtil.killCFG(fixedStateSplit); } } } } } return nonReplaceableFrameState; } private static DeoptimizeNode addDeoptimizeNode(StructuredGraph graph, DeoptimizationAction action, DeoptimizationReason reason) { GraalError.guarantee(graph.getGuardsStage() == GuardsStage.FLOATING_GUARDS, "Cannot introduce speculative deoptimization when Graal is used with fixed guards"); return graph.add(new DeoptimizeNode(action, reason)); }
Ensure that all states are either BytecodeFrame.INVALID_FRAMESTATE_BCI or one of BytecodeFrame.AFTER_BCI or BytecodeFrame.BEFORE_BCI. Mixing of before and after isn't allowed.
/** * Ensure that all states are either {@link BytecodeFrame#INVALID_FRAMESTATE_BCI} or one of * {@link BytecodeFrame#AFTER_BCI} or {@link BytecodeFrame#BEFORE_BCI}. Mixing of before and * after isn't allowed. */
private static boolean checkContainsOnlyInvalidOrAfterFrameState(UnmodifiableEconomicMap<Node, Node> duplicates) { int okBci = BytecodeFrame.INVALID_FRAMESTATE_BCI; for (Node node : duplicates.getValues()) { if (node instanceof FrameState) { FrameState frameState = (FrameState) node; if (frameState.bci == BytecodeFrame.INVALID_FRAMESTATE_BCI) { continue; } if (frameState.bci == BytecodeFrame.AFTER_BCI || frameState.bci == BytecodeFrame.BEFORE_BCI) { if (okBci == BytecodeFrame.INVALID_FRAMESTATE_BCI) { okBci = frameState.bci; } else { assert okBci == frameState.bci : node.toString(Verbosity.Debugger); } } else { assert false : node.toString(Verbosity.Debugger); } } } return true; }
Gets the receiver for an invoke, adding a guard if necessary to ensure it is non-null, and ensuring that the resulting type is compatible with the method being invoked.
/** * Gets the receiver for an invoke, adding a guard if necessary to ensure it is non-null, and * ensuring that the resulting type is compatible with the method being invoked. */
@SuppressWarnings("try") public static ValueNode nonNullReceiver(Invoke invoke) { try (DebugCloseable position = invoke.asNode().withNodeSourcePosition()) { MethodCallTargetNode callTarget = (MethodCallTargetNode) invoke.callTarget(); assert !callTarget.isStatic() : callTarget.targetMethod(); StructuredGraph graph = callTarget.graph(); ValueNode oldReceiver = callTarget.arguments().get(0); ValueNode newReceiver = oldReceiver; if (newReceiver.getStackKind() == JavaKind.Object) { if (invoke.getInvokeKind() == InvokeKind.Special) { Stamp paramStamp = newReceiver.stamp(NodeView.DEFAULT); Stamp stamp = paramStamp.join(StampFactory.object(TypeReference.create(graph.getAssumptions(), callTarget.targetMethod().getDeclaringClass()))); if (!stamp.equals(paramStamp)) { // The verifier and previous optimizations guarantee unconditionally that // the // receiver is at least of the type of the method holder for a special // invoke. newReceiver = graph.unique(new PiNode(newReceiver, stamp)); } } if (!StampTool.isPointerNonNull(newReceiver)) { LogicNode condition = graph.unique(IsNullNode.create(newReceiver)); FixedGuardNode fixedGuard = graph.add(new FixedGuardNode(condition, NullCheckException, InvalidateReprofile, true)); PiNode nonNullReceiver = graph.unique(new PiNode(newReceiver, StampFactory.objectNonNull(), fixedGuard)); graph.addBeforeFixed(invoke.asNode(), fixedGuard); newReceiver = nonNullReceiver; } } if (newReceiver != oldReceiver) { callTarget.replaceFirstInput(oldReceiver, newReceiver); } return newReceiver; } } public static boolean canIntrinsify(Replacements replacements, ResolvedJavaMethod target, int invokeBci) { return replacements.hasSubstitution(target, invokeBci); } public static StructuredGraph getIntrinsicGraph(Replacements replacements, ResolvedJavaMethod target, int invokeBci, boolean trackNodeSourcePosition, NodeSourcePosition replaceePosition) { return replacements.getSubstitution(target, invokeBci, trackNodeSourcePosition, replaceePosition); } public static FixedWithNextNode inlineMacroNode(Invoke invoke, ResolvedJavaMethod concrete, Class<? extends FixedWithNextNode> macroNodeClass) throws GraalError { StructuredGraph graph = invoke.asNode().graph(); if (!concrete.equals(((MethodCallTargetNode) invoke.callTarget()).targetMethod())) { assert ((MethodCallTargetNode) invoke.callTarget()).invokeKind().hasReceiver(); InliningUtil.replaceInvokeCallTarget(invoke, graph, InvokeKind.Special, concrete); } FixedWithNextNode macroNode = createMacroNodeInstance(macroNodeClass, invoke); CallTargetNode callTarget = invoke.callTarget(); if (invoke instanceof InvokeNode) { graph.replaceFixedWithFixed((InvokeNode) invoke, graph.add(macroNode)); } else { InvokeWithExceptionNode invokeWithException = (InvokeWithExceptionNode) invoke; invokeWithException.killExceptionEdge(); graph.replaceSplitWithFixed(invokeWithException, graph.add(macroNode), invokeWithException.next()); } GraphUtil.killWithUnusedFloatingInputs(callTarget); return macroNode; } private static FixedWithNextNode createMacroNodeInstance(Class<? extends FixedWithNextNode> macroNodeClass, Invoke invoke) throws GraalError { try { Constructor<?> cons = macroNodeClass.getDeclaredConstructor(Invoke.class); return (FixedWithNextNode) cons.newInstance(invoke); } catch (ReflectiveOperationException | IllegalArgumentException | SecurityException e) { throw new GraalGraphError(e).addContext(invoke.asNode()).addContext("macroSubstitution", macroNodeClass); } }
This method exclude InstrumentationNode from inlining heuristics.
/** * This method exclude InstrumentationNode from inlining heuristics. */
public static int getNodeCount(StructuredGraph graph) { return graph.getNodeCount(); } }