org.pkl.thirdparty.truffle.api.profiles.ConditionProfile Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of pkl-tools Show documentation
Show all versions of pkl-tools Show documentation
Fat Jar containing pkl-cli, pkl-codegen-java, pkl-codegen-kotlin, pkl-config-java, pkl-core, pkl-doc, and their shaded third-party dependencies.
/*
* Copyright (c) 2014, 2020, Oracle and/or its affiliates. All rights reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* The Universal Permissive License (UPL), Version 1.0
*
* Subject to the condition set forth below, permission is hereby granted to any
* person obtaining a copy of this software, associated documentation and/or
* data (collectively the "Software"), free of charge and under any and all
* copyright rights in the Software, and any and all patent rights owned or
* freely licensable by each licensor hereunder covering either (i) the
* unmodified Software as contributed to or provided by such licensor, or (ii)
* the Larger Works (as defined below), to deal in both
*
* (a) the Software, and
*
* (b) any piece of software and/or hardware listed in the lrgrwrks.txt file if
* one is included with the Software each a "Larger Work" to which the Software
* is contributed by such licensors),
*
* without restriction, including without limitation the rights to copy, create
* derivative works of, display, perform, and distribute the Software and make,
* use, sell, offer for sale, import, export, have made, and have sold the
* Software and the Larger Work(s), and to sublicense the foregoing rights on
* either these or other terms.
*
* This license is subject to the following condition:
*
* The above copyright notice and either this complete permission notice or at a
* minimum a reference to the UPL must be included in all copies or substantial
* portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
package org.pkl.thirdparty.truffle.api.profiles;
import org.pkl.thirdparty.truffle.api.CompilerDirectives;
import org.pkl.thirdparty.truffle.api.CompilerDirectives.CompilationFinal;
import org.pkl.thirdparty.truffle.api.dsl.InlineSupport.InlineTarget;
import org.pkl.thirdparty.truffle.api.dsl.NeverDefault;
/**
*
* ConditionProfiles are useful to profile the outcome of conditions. A regular condition profile
* keeps track of a binary state, for each branch whether a branch was hit or not and communicates
* this to the compiler. If frequency information for each branch should be collected use
* {@link CountingConditionProfile} instead.
*
*
*
* Usage example:
*
*
* class AbsoluteNode extends Node {
*
* final ConditionProfile greaterZeroProfile = ConditionProfile.create();
*
* void execute(int value) {
* if (greaterZeroProfile.profile(value >= 0)) {
* return value;
* } else {
* return -value;
* }
* }
* }
*
*
* {@inheritDoc}
*
* @see #create()
* @see LoopConditionProfile
* @see CountingConditionProfile
* @since 0.10
*/
public class ConditionProfile extends Profile {
private static final ConditionProfile DISABLED;
static {
ConditionProfile profile = new ConditionProfile();
profile.disable();
DISABLED = profile;
}
ConditionProfile() {
}
@CompilationFinal private boolean wasTrue;
@CompilationFinal private boolean wasFalse;
/** @since 0.10 */
public boolean profile(boolean value) {
if (value) {
if (!wasTrue) {
CompilerDirectives.transferToInterpreterAndInvalidate();
wasTrue = true;
}
return true;
} else {
if (!wasFalse) {
CompilerDirectives.transferToInterpreterAndInvalidate();
wasFalse = true;
}
return false;
}
}
/**
* {@inheritDoc}
*
* @since 23.0
*/
@Override
public void disable() {
this.wasFalse = true;
this.wasTrue = true;
}
/**
* {@inheritDoc}
*
* @since 23.0
*/
@Override
public void reset() {
if (this == DISABLED) {
return;
}
this.wasFalse = false;
this.wasTrue = false;
}
boolean wasTrue() {
return wasTrue;
}
boolean wasFalse() {
return wasFalse;
}
/**
* {@inheritDoc}
*
* @since 23.0
*/
@Override
public String toString() {
if (this == DISABLED) {
return toStringDisabled();
} else {
return String.format("%s(wasTrue=%s, wasFalse=%s)@%x", getClass().getSimpleName(), wasTrue, wasFalse, hashCode());
}
}
/**
* @since 0.10
* @deprecated use {@link CountingConditionProfile} instead
*/
@Deprecated
public static ConditionProfile createCountingProfile() {
if (isProfilingEnabled()) {
return Counting.createLazyLoadClass();
} else {
return Disabled.INSTANCE;
}
}
/**
* @since 0.10
* @deprecated use {@link ConditionProfile#create()} instead.
*/
@Deprecated
@NeverDefault
public static ConditionProfile createBinaryProfile() {
if (isProfilingEnabled()) {
return new ConditionProfile();
} else {
return DISABLED;
}
}
/**
* Returns a {@link ConditionProfile} that speculates on conditions to be never
* true or to be never false. Condition profiles are intended to be
* used as part of if conditions.
*
* @since 20.2
*/
@NeverDefault
public static ConditionProfile create() {
return createBinaryProfile();
}
/**
* Returns the uncached version of the profile. The uncached version of a profile does nothing.
*
* @since 19.0
*/
public static ConditionProfile getUncached() {
return DISABLED;
}
/**
* Returns an inlined version of the profile. This version is automatically used by Truffle DSL
* node inlining.
*
* @since 23.0
*/
public static InlinedConditionProfile inline(InlineTarget target) {
return InlinedConditionProfile.inline(target);
}
static final class Disabled extends ConditionProfile {
static final ConditionProfile INSTANCE = new Disabled();
@Override
protected Object clone() {
return INSTANCE;
}
@Override
public boolean profile(boolean value) {
return value;
}
@Override
public String toString() {
return toStringDisabled();
}
}
/*
* Code to be removed with deprecated API. New code lives in CountingConditionProfile.
*/
static final class Counting extends ConditionProfile {
@CompilationFinal private int trueCount;
@CompilationFinal private int falseCount;
/**
* A constant holding the maximum value an {@code int} can have, 230-1. The sum
* of the true and false count must not overflow. This constant is used to check whether one
* of the counts does not exceed the required maximum value.
*/
public static final int MAX_VALUE = 0x3fffffff;
Counting() {
}
@Override
public boolean profile(boolean value) {
// locals required to guarantee no overflow in multi-threaded environments
int t = trueCount;
int f = falseCount;
boolean val = value;
if (val) {
if (t == 0) {
CompilerDirectives.transferToInterpreterAndInvalidate();
}
if (f == 0) {
// Make this branch fold during PE
val = true;
}
if (CompilerDirectives.inInterpreter()) {
if (t < MAX_VALUE) {
trueCount = t + 1;
}
}
} else {
if (f == 0) {
CompilerDirectives.transferToInterpreterAndInvalidate();
}
if (t == 0) {
// Make this branch fold during PE
val = false;
}
if (CompilerDirectives.inInterpreter()) {
if (f < MAX_VALUE) {
falseCount = f + 1;
}
}
}
if (CompilerDirectives.inInterpreter()) {
// no branch probability calculation in the interpreter
return val;
} else {
int sum = t + f;
return CompilerDirectives.injectBranchProbability((double) t / (double) sum, val);
}
}
@Override
public void disable() {
if (this.trueCount == 0) {
this.trueCount = 1;
}
if (this.falseCount == 0) {
this.falseCount = 1;
}
}
@Override
public void reset() {
this.trueCount = 0;
this.falseCount = 0;
}
int getTrueCount() {
return trueCount;
}
int getFalseCount() {
return falseCount;
}
@Override
public String toString() {
int t = trueCount;
int f = falseCount;
int sum = t + f;
String details = String.format("trueProbability=%s (trueCount=%s, falseCount=%s)", (double) t / (double) sum, t, f);
return toString(ConditionProfile.class, sum == 0, false, details);
}
/* Needed for lazy class loading. */
static ConditionProfile createLazyLoadClass() {
return new Counting();
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy