Mercurial > hg > truffle
view graal/com.oracle.graal.compiler.test/src/com/oracle/graal/compiler/test/GraalCompilerTest.java @ 15949:4243a6b8dd19
Fix insertion of profile data in unit tests.
author | Roland Schatz <roland.schatz@oracle.com> |
---|---|
date | Wed, 28 May 2014 12:17:09 +0200 |
parents | 5c73b162eec2 |
children | e4567f9acc42 |
line wrap: on
line source
/* * Copyright (c) 2011, 2014, 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 com.oracle.graal.compiler.test; import static com.oracle.graal.api.code.CodeUtil.*; import static com.oracle.graal.compiler.GraalCompiler.*; import static com.oracle.graal.compiler.common.GraalOptions.*; import static com.oracle.graal.nodes.ConstantNode.*; import java.lang.reflect.*; import java.util.*; import java.util.concurrent.atomic.*; import org.junit.*; import org.junit.internal.*; import com.oracle.graal.api.code.*; import com.oracle.graal.api.code.CallingConvention.Type; import com.oracle.graal.api.meta.*; import com.oracle.graal.api.replacements.*; import com.oracle.graal.api.runtime.*; import com.oracle.graal.baseline.*; import com.oracle.graal.compiler.common.*; import com.oracle.graal.compiler.target.*; import com.oracle.graal.debug.*; import com.oracle.graal.debug.Debug.Scope; import com.oracle.graal.graph.*; import com.oracle.graal.graph.Node.Verbosity; import com.oracle.graal.java.*; import com.oracle.graal.lir.asm.*; import com.oracle.graal.nodes.*; import com.oracle.graal.nodes.cfg.*; import com.oracle.graal.nodes.spi.*; import com.oracle.graal.nodes.virtual.*; import com.oracle.graal.phases.*; import com.oracle.graal.phases.common.inlining.*; import com.oracle.graal.phases.schedule.*; import com.oracle.graal.phases.tiers.*; import com.oracle.graal.phases.util.*; import com.oracle.graal.printer.*; import com.oracle.graal.runtime.*; import com.oracle.graal.test.*; /** * Base class for Graal compiler unit tests. * <p> * White box tests for Graal compiler transformations use this pattern: * <ol> * <li>Create a graph by {@linkplain #parse(String) parsing} a method.</li> * <li>Manually modify the graph (e.g. replace a parameter node with a constant).</li> * <li>Apply a transformation to the graph.</li> * <li>Assert that the transformed graph is equal to an expected graph.</li> * </ol> * <p> * See {@link InvokeHintsTest} as an example of a white box test. * <p> * Black box tests use the {@link #test(String, Object...)} or * {@link #testN(int, String, Object...)} to execute some method in the interpreter and compare its * result against that produced by a Graal compiled version of the method. * <p> * These tests will be run by the {@code mx unittest} command. */ public abstract class GraalCompilerTest extends GraalTest { private final Providers providers; private final Backend backend; private final Suites suites; private static boolean substitutionsInstalled; private void installSubstitutions() { if (!substitutionsInstalled) { this.providers.getReplacements().registerSubstitutions(GraalCompilerTest.class, InjectProfileDataSubstitutions.class); substitutionsInstalled = true; } } protected Suites createSuites() { Suites ret = backend.getSuites().createSuites(); ret.getHighTier().findPhase(InliningPhase.class).add(new Phase("ComputeLoopFrequenciesPhase") { @Override protected void run(StructuredGraph graph) { ComputeLoopFrequenciesClosure.compute(graph); } }); return ret; } public GraalCompilerTest() { this.backend = Graal.getRequiredCapability(RuntimeProvider.class).getHostBackend(); this.providers = getBackend().getProviders(); this.suites = createSuites(); installSubstitutions(); } /** * Set up a test for a non-default backend. The test should check (via {@link #getBackend()} ) * whether the desired backend is available. * * @param arch the name of the desired backend architecture */ public GraalCompilerTest(Class<? extends Architecture> arch) { RuntimeProvider runtime = Graal.getRequiredCapability(RuntimeProvider.class); Backend b = runtime.getBackend(arch); if (b != null) { this.backend = b; } else { // Fall back to the default/host backend this.backend = runtime.getHostBackend(); } this.providers = backend.getProviders(); this.suites = createSuites(); installSubstitutions(); } @BeforeClass public static void initializeDebugging() { DebugEnvironment.initialize(System.out); } private Scope debugScope; @Before public void beforeTest() { assert debugScope == null; debugScope = Debug.scope(getClass()); } @After public void afterTest() { if (debugScope != null) { debugScope.close(); } debugScope = null; } protected void assertEquals(StructuredGraph expected, StructuredGraph graph) { assertEquals(expected, graph, false, true); } protected int countUnusedConstants(StructuredGraph graph) { int total = 0; for (ConstantNode node : getConstantNodes(graph)) { if (!ConstantNodeRecordsUsages) { if (node.gatherUsages(graph).isEmpty()) { total++; } } else { if (node.usages().isEmpty()) { total++; } } } return total; } protected int getNodeCountExcludingUnusedConstants(StructuredGraph graph) { return graph.getNodeCount() - countUnusedConstants(graph); } protected void assertEquals(StructuredGraph expected, StructuredGraph graph, boolean excludeVirtual, boolean checkConstants) { String expectedString = getCanonicalGraphString(expected, excludeVirtual, checkConstants); String actualString = getCanonicalGraphString(graph, excludeVirtual, checkConstants); String mismatchString = "mismatch in graphs:\n========= expected (" + expected + ") =========\n" + expectedString + "\n\n========= actual (" + graph + ") =========\n" + actualString; if (!excludeVirtual && getNodeCountExcludingUnusedConstants(expected) != getNodeCountExcludingUnusedConstants(graph)) { Debug.dump(expected, "Node count not matching - expected"); Debug.dump(graph, "Node count not matching - actual"); Assert.fail("Graphs do not have the same number of nodes: " + expected.getNodeCount() + " vs. " + graph.getNodeCount() + "\n" + mismatchString); } if (!expectedString.equals(actualString)) { Debug.dump(expected, "mismatching graphs - expected"); Debug.dump(graph, "mismatching graphs - actual"); Assert.fail(mismatchString); } } protected void assertConstantReturn(StructuredGraph graph, int value) { String graphString = getCanonicalGraphString(graph, false, true); Assert.assertEquals("unexpected number of ReturnNodes: " + graphString, graph.getNodes(ReturnNode.class).count(), 1); ValueNode result = graph.getNodes(ReturnNode.class).first().result(); Assert.assertTrue("unexpected ReturnNode result node: " + graphString, result.isConstant()); Assert.assertEquals("unexpected ReturnNode result kind: " + graphString, result.asConstant().getKind(), Kind.Int); Assert.assertEquals("unexpected ReturnNode result: " + graphString, result.asConstant().asInt(), value); } protected static String getCanonicalGraphString(StructuredGraph graph, boolean excludeVirtual, boolean checkConstants) { SchedulePhase schedule = new SchedulePhase(); schedule.apply(graph); NodeMap<Integer> canonicalId = graph.createNodeMap(); int nextId = 0; StringBuilder result = new StringBuilder(); for (Block block : schedule.getCFG().getBlocks()) { result.append("Block " + block + " "); if (block == schedule.getCFG().getStartBlock()) { result.append("* "); } result.append("-> "); for (Block succ : block.getSuccessors()) { result.append(succ + " "); } result.append("\n"); for (Node node : schedule.getBlockToNodesMap().get(block)) { if (node.recordsUsages()) { if (!excludeVirtual || !(node instanceof VirtualObjectNode || node instanceof ProxyNode)) { int id; if (canonicalId.get(node) != null) { id = canonicalId.get(node); } else { id = nextId++; canonicalId.set(node, id); } String name = node instanceof ConstantNode && checkConstants ? node.toString(Verbosity.Name) : node.getClass().getSimpleName(); result.append(" " + id + "|" + name + (excludeVirtual ? "\n" : " (" + node.usages().count() + ")\n")); } } } } return result.toString(); } protected Backend getBackend() { return backend; } protected Suites getSuites() { return suites; } protected Providers getProviders() { return providers; } protected SnippetReflectionProvider getSnippetReflection() { return Graal.getRequiredCapability(SnippetReflectionProvider.class); } protected TargetDescription getTarget() { return getProviders().getCodeCache().getTarget(); } protected CodeCacheProvider getCodeCache() { return getProviders().getCodeCache(); } protected ConstantReflectionProvider getConstantReflection() { return getProviders().getConstantReflection(); } protected MetaAccessProvider getMetaAccess() { return getProviders().getMetaAccess(); } protected LoweringProvider getLowerer() { return getProviders().getLowerer(); } /** * Parses a Java method to produce a graph. * * @param methodName the name of the method in {@code this.getClass()} to be parsed */ protected StructuredGraph parse(String methodName) { return parse(getMethod(methodName)); } private static AtomicInteger compilationId = new AtomicInteger(); protected void testN(int n, final String name, final Object... args) { final List<Throwable> errors = new ArrayList<>(n); Thread[] threads = new Thread[n]; for (int i = 0; i < n; i++) { Thread t = new Thread(i + ":" + name) { @Override public void run() { try { test(name, args); } catch (Throwable e) { errors.add(e); } } }; threads[i] = t; t.start(); } for (int i = 0; i < n; i++) { try { threads[i].join(); } catch (InterruptedException e) { errors.add(e); } } if (!errors.isEmpty()) { throw new MultiCauseAssertionError(errors.size() + " failures", errors.toArray(new Throwable[errors.size()])); } } protected Object referenceInvoke(Method method, Object receiver, Object... args) throws IllegalAccessException, IllegalArgumentException, InvocationTargetException { return method.invoke(receiver, args); } protected static class Result { final Object returnValue; final Throwable exception; public Result(Object returnValue, Throwable exception) { this.returnValue = returnValue; this.exception = exception; } @Override public String toString() { return exception == null ? returnValue == null ? "null" : returnValue.toString() : "!" + exception; } } /** * Called before a test is executed. */ protected void before(@SuppressWarnings("unused") Method method) { } /** * Called after a test is executed. */ protected void after() { } protected Result executeExpected(Method method, Object receiver, Object... args) { before(method); try { // This gives us both the expected return value as well as ensuring that the method to // be compiled is fully resolved return new Result(referenceInvoke(method, receiver, args), null); } catch (InvocationTargetException e) { return new Result(null, e.getTargetException()); } catch (Exception e) { throw new RuntimeException(e); } finally { after(); } } protected Result executeActual(Method method, Object receiver, Object... args) { before(method); Object[] executeArgs = argsWithReceiver(receiver, args); ResolvedJavaMethod javaMethod = getMetaAccess().lookupJavaMethod(method); checkArgs(javaMethod, executeArgs); InstalledCode compiledMethod = null; if (UseBaselineCompiler.getValue()) { compiledMethod = getCodeBaseline(javaMethod, method); } else { compiledMethod = getCode(javaMethod, parse(method)); } try { return new Result(compiledMethod.executeVarargs(executeArgs), null); } catch (Throwable e) { return new Result(null, e); } finally { after(); } } protected InstalledCode getCodeBaseline(ResolvedJavaMethod javaMethod, Method method) { return getCodeBaseline(javaMethod, method, false); } protected InstalledCode getCodeBaseline(ResolvedJavaMethod javaMethod, Method method, boolean forceCompile) { assert method.getAnnotation(Test.class) == null : "shouldn't parse method with @Test annotation: " + method; try (Scope bds = Debug.scope("Baseline")) { Debug.log("getCodeBaseline()"); } catch (Throwable e) { throw Debug.handle(e); } if (!forceCompile) { InstalledCode cached = cache.get(javaMethod); if (cached != null) { if (cached.isValid()) { return cached; } } } final int id = compilationId.incrementAndGet(); InstalledCode installedCode = null; try (Scope ds = Debug.scope("Compiling", new DebugDumpScope(String.valueOf(id), true))) { final boolean printCompilation = PrintCompilation.getValue() && !TTY.isSuppressed(); if (printCompilation) { TTY.println(String.format("@%-6d Graal %-70s %-45s %-50s ...", id, javaMethod.getDeclaringClass().getName(), javaMethod.getName(), javaMethod.getSignature())); } long start = System.currentTimeMillis(); CompilationResult compResult = compileBaseline(javaMethod); if (printCompilation) { TTY.println(String.format("@%-6d Graal %-70s %-45s %-50s | %4dms %5dB", id, "", "", "", System.currentTimeMillis() - start, compResult.getTargetCodeSize())); } try (Scope s = Debug.scope("CodeInstall", getCodeCache(), javaMethod)) { installedCode = addMethod(javaMethod, compResult); if (installedCode == null) { throw new GraalInternalError("Could not install code for " + MetaUtil.format("%H.%n(%p)", javaMethod)); } } catch (Throwable e) { throw Debug.handle(e); } } catch (Throwable e) { throw Debug.handle(e); } if (!forceCompile) { cache.put(javaMethod, installedCode); } return installedCode; } private CompilationResult compileBaseline(ResolvedJavaMethod javaMethod) { try (Scope bds = Debug.scope("CompileBaseline", javaMethod, providers.getCodeCache())) { BaselineCompiler baselineCompiler = new BaselineCompiler(GraphBuilderConfiguration.getDefault(), providers.getMetaAccess()); OptimisticOptimizations optimisticOpts = OptimisticOptimizations.ALL; return baselineCompiler.generate(javaMethod, -1, getBackend(), new CompilationResult(), javaMethod, CompilationResultBuilderFactory.Default, optimisticOpts); } catch (Throwable e) { throw Debug.handle(e); } } protected void checkArgs(ResolvedJavaMethod method, Object[] args) { JavaType[] sig = MetaUtil.signatureToTypes(method); Assert.assertEquals(sig.length, args.length); for (int i = 0; i < args.length; i++) { JavaType javaType = sig[i]; Kind kind = javaType.getKind(); Object arg = args[i]; if (kind == Kind.Object) { if (arg != null && javaType instanceof ResolvedJavaType) { ResolvedJavaType resolvedJavaType = (ResolvedJavaType) javaType; Assert.assertTrue(resolvedJavaType + " from " + getMetaAccess().lookupJavaType(arg.getClass()), resolvedJavaType.isAssignableFrom(getMetaAccess().lookupJavaType(arg.getClass()))); } } else { Assert.assertNotNull(arg); Assert.assertEquals(kind.toBoxedJavaClass(), arg.getClass()); } } } /** * Prepends a non-null receiver argument to a given list or args. * * @param receiver the receiver argument to prepend if it is non-null */ protected Object[] argsWithReceiver(Object receiver, Object... args) { Object[] executeArgs; if (receiver == null) { executeArgs = args; } else { executeArgs = new Object[args.length + 1]; executeArgs[0] = receiver; for (int i = 0; i < args.length; i++) { executeArgs[i + 1] = args[i]; } } return executeArgs; } protected void test(String name, Object... args) { try { Method method = getMethod(name); Object receiver = Modifier.isStatic(method.getModifiers()) ? null : this; test(method, receiver, args); } catch (AssumptionViolatedException e) { // Suppress so that subsequent calls to this method within the // same Junit @Test annotated method can proceed. } } protected void test(Method method, Object receiver, Object... args) { Result expect = executeExpected(method, receiver, args); if (getCodeCache() == null) { return; } testAgainstExpected(method, expect, receiver, args); } protected void testAgainstExpected(Method method, Result expect, Object receiver, Object... args) { testAgainstExpected(method, expect, Collections.<DeoptimizationReason> emptySet(), receiver, args); } protected Result executeActualCheckDeopt(Method method, Set<DeoptimizationReason> shouldNotDeopt, Object receiver, Object... args) { Map<DeoptimizationReason, Integer> deoptCounts = new EnumMap<>(DeoptimizationReason.class); ResolvedJavaMethod javaMethod = getMetaAccess().lookupJavaMethod(method); ProfilingInfo profile = javaMethod.getProfilingInfo(); for (DeoptimizationReason reason : shouldNotDeopt) { deoptCounts.put(reason, profile.getDeoptimizationCount(reason)); } Result actual = executeActual(method, receiver, args); profile = javaMethod.getProfilingInfo(); // profile can change after execution for (DeoptimizationReason reason : shouldNotDeopt) { Assert.assertEquals((int) deoptCounts.get(reason), profile.getDeoptimizationCount(reason)); } return actual; } protected void assertEquals(Result expect, Result actual) { if (expect.exception != null) { Assert.assertTrue("expected " + expect.exception, actual.exception != null); Assert.assertEquals("Exception class", expect.exception.getClass(), actual.exception.getClass()); Assert.assertEquals("Exception message", expect.exception.getMessage(), actual.exception.getMessage()); } else { if (actual.exception != null) { actual.exception.printStackTrace(); Assert.fail("expected " + expect.returnValue + " but got an exception"); } assertDeepEquals(expect.returnValue, actual.returnValue); } } protected void testAgainstExpected(Method method, Result expect, Set<DeoptimizationReason> shouldNotDeopt, Object receiver, Object... args) { Result actual = executeActualCheckDeopt(method, shouldNotDeopt, receiver, args); assertEquals(expect, actual); } private Map<ResolvedJavaMethod, InstalledCode> cache = new HashMap<>(); /** * Gets installed code for a given method and graph, compiling it first if necessary. */ protected InstalledCode getCode(final ResolvedJavaMethod method, final StructuredGraph graph) { return getCode(method, graph, false); } /** * Gets installed code for a given method and graph, compiling it first if necessary. * * @param forceCompile specifies whether to ignore any previous code cached for the (method, * key) pair */ protected InstalledCode getCode(final ResolvedJavaMethod method, final StructuredGraph graph, boolean forceCompile) { if (!forceCompile) { InstalledCode cached = cache.get(method); if (cached != null) { if (cached.isValid()) { return cached; } } } final int id = compilationId.incrementAndGet(); InstalledCode installedCode = null; try (AllocSpy spy = AllocSpy.open(method); Scope ds = Debug.scope("Compiling", new DebugDumpScope(String.valueOf(id), true))) { final boolean printCompilation = PrintCompilation.getValue() && !TTY.isSuppressed(); if (printCompilation) { TTY.println(String.format("@%-6d Graal %-70s %-45s %-50s ...", id, method.getDeclaringClass().getName(), method.getName(), method.getSignature())); } long start = System.currentTimeMillis(); CompilationResult compResult = compile(method, graph); if (printCompilation) { TTY.println(String.format("@%-6d Graal %-70s %-45s %-50s | %4dms %5dB", id, "", "", "", System.currentTimeMillis() - start, compResult.getTargetCodeSize())); } try (Scope s = Debug.scope("CodeInstall", getCodeCache(), method)) { installedCode = addMethod(method, compResult); if (installedCode == null) { throw new GraalInternalError("Could not install code for " + MetaUtil.format("%H.%n(%p)", method)); } } catch (Throwable e) { throw Debug.handle(e); } } catch (Throwable e) { throw Debug.handle(e); } if (!forceCompile) { cache.put(method, installedCode); } return installedCode; } protected CompilationResult compile(ResolvedJavaMethod method, final StructuredGraph graph) { CallingConvention cc = getCallingConvention(getCodeCache(), Type.JavaCallee, graph.method(), false); return compileGraph(graph, null, cc, method, getProviders(), getBackend(), getCodeCache().getTarget(), null, getDefaultGraphBuilderSuite(), OptimisticOptimizations.ALL, getProfilingInfo(graph), getSpeculationLog(), getSuites(), new CompilationResult(), CompilationResultBuilderFactory.Default); } protected SpeculationLog getSpeculationLog() { return null; } protected InstalledCode addMethod(final ResolvedJavaMethod method, final CompilationResult compResult) { return getCodeCache().addMethod(method, compResult, null, null); } /** * Parses a Java method to produce a graph. * * @param methodName the name of the method in {@code this.getClass()} to be parsed */ protected StructuredGraph parseProfiled(String methodName) { return parseProfiled(getMethod(methodName)); } /** * Parses a Java method to produce a graph. */ protected StructuredGraph parse(Method m) { return parse0(m, getCustomGraphBuilderSuite(GraphBuilderConfiguration.getEagerDefault())); } /** * Parses a Java method to produce a graph. */ protected StructuredGraph parseProfiled(Method m) { return parse0(m, getDefaultGraphBuilderSuite()); } /** * Parses a Java method in debug mode to produce a graph with extra infopoints. */ protected StructuredGraph parseDebug(Method m) { return parse0(m, getCustomGraphBuilderSuite(GraphBuilderConfiguration.getEagerInfopointDefault())); } private StructuredGraph parse0(Method m, PhaseSuite<HighTierContext> graphBuilderSuite) { assert m.getAnnotation(Test.class) == null : "shouldn't parse method with @Test annotation: " + m; ResolvedJavaMethod javaMethod = getMetaAccess().lookupJavaMethod(m); StructuredGraph graph = new StructuredGraph(javaMethod); graphBuilderSuite.apply(graph, new HighTierContext(providers, null, null, graphBuilderSuite, OptimisticOptimizations.ALL)); return graph; } protected PhaseSuite<HighTierContext> getDefaultGraphBuilderSuite() { // defensive copying return backend.getSuites().getDefaultGraphBuilderSuite().copy(); } protected PhaseSuite<HighTierContext> getCustomGraphBuilderSuite(GraphBuilderConfiguration gbConf) { PhaseSuite<HighTierContext> suite = getDefaultGraphBuilderSuite().copy(); ListIterator<BasePhase<? super HighTierContext>> iterator = suite.findPhase(GraphBuilderPhase.class); iterator.remove(); iterator.add(new GraphBuilderPhase(gbConf)); return suite; } protected Replacements getReplacements() { return getProviders().getReplacements(); } /** * Inject a probability for a branch condition into the profiling information of this test case. * * @param p the probability that cond is true * @param cond the condition of the branch * @return cond */ protected static boolean branchProbability(double p, boolean cond) { return cond; } /** * Inject an iteration count for a loop condition into the profiling information of this test * case. * * @param i the iteration count of the loop * @param cond the condition of the loop * @return cond */ protected static boolean iterationCount(double i, boolean cond) { return cond; } }