specialize InitWrapper execution

Author: qunaibit

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

@@ -53,15 +53,16 @@
 import com.oracle.graal.python.nodes.argument.keywords.ExpandKeywordStarargsNode;
 import com.oracle.graal.python.nodes.argument.positional.ExecutePositionalStarargsNode;
 import com.oracle.graal.python.nodes.argument.positional.PositionalArgumentsNode;
-import com.oracle.graal.python.nodes.call.CallNode;
 import com.oracle.graal.python.nodes.call.special.CallBinaryMethodNode;
 import com.oracle.graal.python.nodes.call.special.CallTernaryMethodNode;
+import com.oracle.graal.python.nodes.call.special.CallVarargsMethodNode;
 import com.oracle.graal.python.nodes.object.IsBuiltinClassProfile;
 import com.oracle.graal.python.runtime.GilNode;
 import com.oracle.graal.python.runtime.exception.PException;
 import com.oracle.truffle.api.CompilerDirectives;
 import com.oracle.truffle.api.dsl.Cached;
 import com.oracle.truffle.api.dsl.Cached.Exclusive;
+import com.oracle.truffle.api.dsl.Specialization;
 import com.oracle.truffle.api.interop.ArityException;
 import com.oracle.truffle.api.interop.InteropLibrary;
 import com.oracle.truffle.api.interop.UnsupportedMessageException;
@@ -208,44 +209,48 @@ public InitWrapper(Object delegate) {
             super(delegate);
         }
 
-        @ExportMessage
-        protected int execute(Object[] arguments,
-                        @CachedLibrary("this") PythonNativeWrapperLibrary lib,
-                        @Cached ExecutePositionalStarargsNode.ExecutePositionalStarargsInteropNode posStarargsNode,
-                        @Cached ExpandKeywordStarargsNode expandKwargsNode,
-                        @Exclusive @Cached CallNode callNode,
-                        @Cached ToJavaNode toJavaNode,
-                        @Cached ConditionProfile arityProfile,
-                        @Cached BranchProfile errorProfile,
-                        @Cached TransformExceptionToNativeNode transformExceptionToNativeNode,
-                        @Exclusive @Cached GilNode gil) throws ArityException {
-            boolean mustRelease = gil.acquire();
-            try {
-                if (arityProfile.profile(arguments.length != 3)) {
-                    CompilerDirectives.transferToInterpreterAndInvalidate();
-                    throw ArityException.create(3, 3, arguments.length);
-                }
+        @ExportMessage(name = "execute")
+        static class Execute {
+
+            @Specialization(guards = "arguments.length == 3")
+            static int init(InitWrapper self, Object[] arguments,
+                            @CachedLibrary("self") PythonNativeWrapperLibrary lib,
+                            @Cached ExecutePositionalStarargsNode.ExecutePositionalStarargsInteropNode posStarargsNode,
+                            @Cached ExpandKeywordStarargsNode expandKwargsNode,
+                            @Cached CallVarargsMethodNode callNode,
+                            @Cached ToJavaNode toJavaNode,
+                            @Cached BranchProfile errorProfile,
+                            @Cached TransformExceptionToNativeNode transformExceptionToNativeNode,
+                            @Exclusive @Cached GilNode gil) {
+                boolean mustRelease = gil.acquire();
                 try {
-                    // convert args
-                    Object receiver = toJavaNode.execute(arguments[0]);
-                    Object starArgs = toJavaNode.execute(arguments[1]);
-                    Object kwArgs = toJavaNode.execute(arguments[2]);
-
-                    Object[] starArgsArray = posStarargsNode.executeWithGlobalState(starArgs);
-                    Object[] pArgs = PositionalArgumentsNode.prependArgument(receiver, starArgsArray);
-                    PKeyword[] kwArgsArray = expandKwargsNode.execute(kwArgs);
-                    callNode.execute(null, lib.getDelegate(this), pArgs, kwArgsArray);
-                    return 0;
-                } catch (PException e) {
-                    errorProfile.enter();
-                    transformExceptionToNativeNode.execute(null, e);
-                    return -1;
+                    try {
+                        // convert args
+                        Object receiver = toJavaNode.execute(arguments[0]);
+                        Object starArgs = toJavaNode.execute(arguments[1]);
+                        Object kwArgs = toJavaNode.execute(arguments[2]);
+
+                        Object[] starArgsArray = posStarargsNode.executeWithGlobalState(starArgs);
+                        Object[] pArgs = PositionalArgumentsNode.prependArgument(receiver, starArgsArray);
+                        PKeyword[] kwArgsArray = expandKwargsNode.execute(kwArgs);
+                        callNode.execute(null, lib.getDelegate(self), pArgs, kwArgsArray);
+                        return 0;
+                    } catch (PException e) {
+                        errorProfile.enter();
+                        transformExceptionToNativeNode.execute(null, e);
+                        return -1;
+                    }
+                } finally {
+                    gil.release(mustRelease);
                 }
-            } finally {
-                gil.release(mustRelease);
             }
-        }
 
+            @Specialization(guards = "arguments.length != 3")
+            static int error(@SuppressWarnings("unused") InitWrapper self, Object[] arguments) throws ArityException {
+                throw ArityException.create(3, 3, arguments.length);
+            }
+
+        }
     }
 
     @ExportLibrary(InteropLibrary.class)