Fix: ensure pointer identity for small int/long wrappers in compiled code.

Author: fangerer

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/objects/cext/CExtNodes.java

@@ -351,11 +351,25 @@ static PrimitiveNativeWrapper doIntegerSmall(@SuppressWarnings("unused") CExtCon
             return PrimitiveNativeWrapper.createInt(i);
         }
 
+        /**
+         * Generic integer conversion.<br/>
+         * In the interpreter: we just lookup the cached primitive native wrapper and use this
+         * instance. In compiled code, we do the same but create a fresh copy such that the compiler
+         * always sees a fresh instance. This avoids a phi and certainly a real allocation. Note: it
+         * is important to copy the existing cached wrapper because otherwise pointer equality is
+         * not ensured (see also:
+         * {@link AsPythonObjectBaseNode#mayUsePrimitive(IsPointerNode, PrimitiveNativeWrapper)}).
+         */
         @Specialization(replaces = "doIntegerSmall")
         static PrimitiveNativeWrapper doInteger(@SuppressWarnings("unused") CExtContext cextContext, int i,
                         @Shared("contextRef") @CachedContext(PythonLanguage.class) ContextReference<PythonContext> contextRef) {
-            if (CompilerDirectives.inInterpreter() && CApiGuards.isSmallInteger(i)) {
-                return doIntegerSmall(cextContext, i, contextRef);
+            if (CApiGuards.isSmallInteger(i)) {
+                if (CompilerDirectives.inInterpreter()) {
+                    return doIntegerSmall(cextContext, i, contextRef);
+                } else {
+                    // for explanation: see method doc
+                    return doIntegerSmall(cextContext, i, contextRef).copy();
+                }
             }
             return PrimitiveNativeWrapper.createInt(i);
         }
@@ -373,8 +387,13 @@ static PrimitiveNativeWrapper doLongSmall(@SuppressWarnings("unused") CExtContex
         @Specialization(replaces = "doLongSmall")
         static PrimitiveNativeWrapper doLong(@SuppressWarnings("unused") CExtContext cextContext, long l,
                         @Shared("contextRef") @CachedContext(PythonLanguage.class) ContextReference<PythonContext> contextRef) {
-            if (CompilerDirectives.inInterpreter() && CApiGuards.isSmallLong(l)) {
-                return doLongSmall(cextContext, l, contextRef);
+            if (CApiGuards.isSmallLong(l)) {
+                // for explanation of this construct: see 'ToSulongNode.doInteger'
+                if (CompilerDirectives.inInterpreter()) {
+                    return doLongSmall(cextContext, l, contextRef);
+                } else {
+                    return doLongSmall(cextContext, l, contextRef).copy();
+                }
             }
             return PrimitiveNativeWrapper.createLong(l);
         }
@@ -602,17 +621,27 @@ static PrimitiveNativeWrapper doLongSmall(@SuppressWarnings("unused") CExtContex
         @Specialization(replaces = "doIntegerSmall")
         static PrimitiveNativeWrapper doInteger(CExtContext cextContext, int i,
                         @Shared("contextRef") @CachedContext(PythonLanguage.class) ContextReference<PythonContext> contextRef) {
-            if (CompilerDirectives.inInterpreter() && CApiGuards.isSmallInteger(i)) {
-                return doIntegerSmall(cextContext, i, contextRef);
+            if (CApiGuards.isSmallInteger(i)) {
+                // for explanation of this construct: see 'ToSulongNode.doInteger'
+                if (CompilerDirectives.inInterpreter()) {
+                    return doIntegerSmall(cextContext, i, contextRef);
+                } else {
+                    return doIntegerSmall(cextContext, i, contextRef).copy();
+                }
             }
             return PrimitiveNativeWrapper.createInt(i);
         }
 
         @Specialization(replaces = "doLongSmall")
         static PrimitiveNativeWrapper doLong(CExtContext cextContext, long l,
                         @Shared("contextRef") @CachedContext(PythonLanguage.class) ContextReference<PythonContext> contextRef) {
-            if (CompilerDirectives.inInterpreter() && CApiGuards.isSmallLong(l)) {
-                return doLongSmall(cextContext, l, contextRef);
+            if (CApiGuards.isSmallLong(l)) {
+                // for explanation of this construct: see 'ToSulongNode.doInteger'
+                if (CompilerDirectives.inInterpreter()) {
+                    return doLongSmall(cextContext, l, contextRef);
+                } else {
+                    return doLongSmall(cextContext, l, contextRef).copy();
+                }
             }
             return PrimitiveNativeWrapper.createLong(l);
         }

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/objects/cext/DynamicObjectNativeWrapper.java

@@ -1474,6 +1474,12 @@ private PrimitiveNativeWrapper(double dvalue) {
             this.dvalue = dvalue;
         }
 
+        private PrimitiveNativeWrapper(byte state, long value, double dvalue) {
+            this.state = state;
+            this.value = value;
+            this.dvalue = dvalue;
+        }
+
         public byte getState() {
             return state;
         }
@@ -1553,7 +1559,11 @@ public boolean equals(Object obj) {
 
         @Override
         public int hashCode() {
-            return (int) (value ^ Double.doubleToRawLongBits(dvalue) ^ state);
+            return (Long.hashCode(value) ^ Long.hashCode(Double.doubleToRawLongBits(dvalue)) ^ state);
+        }
+
+        PrimitiveNativeWrapper copy() {
+            return new PrimitiveNativeWrapper(state, value, dvalue);
         }
 
         @Override