[GrCUDA-66] fix arity deprecation (#27)

AlbertoParravicini · web-flow · commit 920f52ca5895 · 2021-12-11T13:32:17.000+01:00
* removed deprecation warning for arity exception

* updated changelog
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,12 +1,12 @@
+# 2021-11-29
+
+* Removed deprecation warning for Truffle's ArityException. 
+
 # 2021-11-21
 
 * Enabled support for cuSPARSE
-  * Operations with sparse matrices from cuSPARSE library are now supported
-  * **Known limitation:** Not all data types are supported: in particular Tgemvi does not support double data types 
-  (both complex and not)
-  * Concurrent operations on parallel streams were analyzed using Nvidia Profiler
-
-=======
+  * Added support for CSR and COO `spmv` and `gemvi`.
+  * **Known limitation:** Tgemvi works only with single-precision floating-point arithmetics.
 
 # 2021-11-17
 
diff --git a/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/cudalibraries/cuml/CUMLRegistry.java b/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/cudalibraries/cuml/CUMLRegistry.java
@@ -128,7 +128,7 @@ public Object call(Object[] arguments) throws ArityException {
                 @TruffleBoundary
                 public Object call(Object[] arguments) throws ArityException {
                     checkArgumentLength(arguments, 2);
-                    try (UnsafeHelper.Integer64Object handle = UnsafeHelper.createInteger64Object()) {
+                    try {
                         long stream_handle = expectLong(arguments[0]);
                         long streamID = expectLong(arguments[1]);
                         Object result = INTEROP.execute(cumlSetStreamFunctionNFI, stream_handle, streamID);
diff --git a/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/functions/DeviceArrayCopyFunction.java b/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/functions/DeviceArrayCopyFunction.java
@@ -102,7 +102,7 @@ Object execute(Object[] arguments,
             numElements = extractNumber(arguments[1], numElementsAccess);
         } else {
             CompilerDirectives.transferToInterpreter();
-            throw ArityException.create(1, arguments.length);
+            throw ArityException.create(1, 2, arguments.length);
         }
         // Obtain what kind of copy (pointer or array) should be executed.
         // By default, see if we can use the fast CUDA memcpy: we cannot use it if the source and target arrays
diff --git a/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/functions/DeviceArrayFunction.java b/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/functions/DeviceArrayFunction.java
@@ -77,7 +77,7 @@ public DeviceArrayFunction(AbstractGrCUDAExecutionContext grCUDAExecutionContext
     @TruffleBoundary
     public Object call(Object[] arguments) throws ArityException, UnsupportedTypeException {
         if (arguments.length < 1) {
-            throw ArityException.create(1, arguments.length);
+            throw ArityException.create(1, 2, arguments.length);
         }
         String typeName = expectString(arguments[0], "first argument of DeviceArray must be string (type name)");
         Type elementType;
diff --git a/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/functions/Function.java b/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/functions/Function.java
@@ -112,7 +112,14 @@ public static long expectPositiveLong(Object number) throws UnsupportedTypeExcep
     public static void checkArgumentLength(Object[] arguments, int expected) throws ArityException {
         if (arguments.length != expected) {
             CompilerDirectives.transferToInterpreter();
-            throw ArityException.create(expected, arguments.length);
+            throw ArityException.create(expected, expected, arguments.length);
+        }
+    }
+
+    public static void checkArgumentLength(Object[] arguments, int minExpected, int maxExpected) throws ArityException {
+        if (arguments.length < minExpected || arguments.length > maxExpected) {
+            CompilerDirectives.transferToInterpreter();
+            throw ArityException.create(minExpected, maxExpected, arguments.length);
         }
     }
 
diff --git a/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/functions/MapDeviceArrayFunction.java b/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/functions/MapDeviceArrayFunction.java
@@ -227,7 +227,7 @@ public Object execute(Object[] arguments,
                     @Cached MapArrayNode mapNode) throws ArityException, UnsupportedTypeException {
         if (arguments.length < 1) {
             CompilerDirectives.transferToInterpreter();
-            throw ArityException.create(1, arguments.length);
+            throw ArityException.create(1, 2, arguments.length);
         }
         String typeName;
         try {
@@ -244,12 +244,11 @@ public Object execute(Object[] arguments,
         }
         if (arguments.length == 1) {
             return new TypedMapDeviceArrayFunction(grCUDAExecutionContext, elementType);
-        } else {
-            if (arguments.length != 2) {
-                CompilerDirectives.transferToInterpreter();
-                throw ArityException.create(2, arguments.length);
-            }
+        } else if (arguments.length == 2) {
             return mapNode.execute(arguments[1], elementType, grCUDAExecutionContext);
+        } else {
+            CompilerDirectives.transferToInterpreter();
+            throw ArityException.create(1, 2, arguments.length);
         }
     }
 }
diff --git a/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/functions/TypedDeviceArrayFunction.java b/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/functions/TypedDeviceArrayFunction.java
@@ -61,7 +61,9 @@ public TypedDeviceArrayFunction(AbstractGrCUDAExecutionContext grCUDAExecutionCo
     public Object call(Object[] arguments) throws ArityException, UnsupportedTypeException {
         if (arguments.length < 1) {
             CompilerDirectives.transferToInterpreter();
-            throw ArityException.create(1, arguments.length);
+            // FIXME: the maximum number of arguments is unbound (as each argument is a dimension of a N-dimensional tensor).
+            //  Truffle currently uses -1 to handle an unbound number of arguments;
+            throw ArityException.create(1, -1, arguments.length);
         }
         return DeviceArrayFunction.createArray(arguments, 0, elementType, grCUDAExecutionContext);
     }
diff --git a/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/functions/TypedMapDeviceArrayFunction.java b/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/functions/TypedMapDeviceArrayFunction.java
@@ -64,7 +64,7 @@ public Object execute(Object[] arguments,
                     @Cached MapArrayNode mapNode) throws ArityException {
         if (arguments.length != 1) {
             CompilerDirectives.transferToInterpreter();
-            throw ArityException.create(1, arguments.length);
+            throw ArityException.create(1, 1, arguments.length);
         }
         return mapNode.execute(arguments[0], elementType, grCUDAExecutionContext);
     }
diff --git a/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/functions/map/MapFunction.java b/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/functions/map/MapFunction.java
@@ -120,7 +120,7 @@ public MappedFunction map(Object function, Object... arguments) throws ArityExce
     static void checkArity(Object[] arguments, int expectedArity) throws ArityException {
         if (arguments.length != expectedArity) {
             CompilerDirectives.transferToInterpreter();
-            throw ArityException.create(expectedArity, arguments.length);
+            throw ArityException.create(expectedArity, expectedArity, arguments.length);
         }
     }
 
@@ -175,7 +175,9 @@ static MapFunctionBase readMemberArg(MapFunction receiver, String member) {
         static MapFunctionBase readMemberSize(MapFunction receiver, String member) {
             return new MapFunctionBase(arguments -> {
                 if (arguments.length == 0) {
-                    throw ArityException.create(1, 0);
+                    // FIXME: the maximum number of arguments is unbound.
+                    //  Truffle currently uses -1 to handle an unbound number of arguments;
+                    throw ArityException.create(1, -1, 0);
                 }
                 try {
                     return receiver.size((MapArgObject) arguments[0], Arrays.copyOfRange(arguments, 1, arguments.length, MapArgObject[].class));
@@ -189,7 +191,9 @@ static MapFunctionBase readMemberSize(MapFunction receiver, String member) {
         static MapFunctionBase readMemberValue(MapFunction receiver, String member) {
             return new MapFunctionBase(arguments -> {
                 if (arguments.length < 1) {
-                    throw ArityException.create(1, arguments.length);
+                    // FIXME: the maximum number of arguments is unbound.
+                    //  Truffle currently uses -1 to handle an unbound number of arguments;
+                    throw ArityException.create(1, -1, arguments.length);
                 }
                 String name = checkString(arguments[0], "name of created value expected");
                 if (arguments.length == 1) {
@@ -225,7 +229,9 @@ Object invokeMember(String member, Object[] arguments,
     @TruffleBoundary
     public MappedFunction execute(Object[] arguments) throws ArityException, UnsupportedTypeException {
         if (arguments.length == 0) {
-            throw ArityException.create(1, 0);
+            // FIXME: the maximum number of arguments is unbound.
+            //  Truffle currently uses -1 to handle an unbound number of arguments;
+            throw ArityException.create(1, -1, 0);
         }
         Object function = arguments[0];
         if (!INTEROP.isExecutable(function)) {
diff --git a/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/functions/map/MappedFunction.java b/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/functions/map/MappedFunction.java
@@ -236,7 +236,7 @@ long execute(Object[] arguments,
                     @Cached("createEqualityProfile()") PrimitiveValueProfile lengthProfile) throws ArityException {
         if (arguments.length != 3) {
             CompilerDirectives.transferToInterpreter();
-            throw ArityException.create(3, arguments.length);
+            throw ArityException.create(3, 3, arguments.length);
         }
         long size;
         try {
@@ -355,7 +355,7 @@ Object execute(Object[] arguments,
                     @Cached("createEqualityProfile()") PrimitiveValueProfile lengthProfile) throws ArityException {
         if (arguments.length != 3) {
             CompilerDirectives.transferToInterpreter();
-            throw ArityException.create(3, arguments.length);
+            throw ArityException.create(3, 3, arguments.length);
         }
         int length = lengthProfile.profile(args.length);
         Object[] mappedArgs = new Object[length];
@@ -443,7 +443,7 @@ Object execute(Object[] arguments,
                     @CachedLibrary(limit = "2") InteropLibrary memberInterop) throws ArityException, UnsupportedTypeException, UnsupportedMessageException {
         if (arguments.length != 3) {
             CompilerDirectives.transferToInterpreter();
-            throw ArityException.create(3, arguments.length);
+            throw ArityException.create(3, 3, arguments.length);
         }
         Object value = parentInterop.execute(parent, arguments);
         try {
@@ -503,7 +503,7 @@ Object execute(Object[] arguments,
                     @CachedLibrary(limit = "2") InteropLibrary elementInterop) throws ArityException, UnsupportedTypeException, UnsupportedMessageException {
         if (arguments.length != 3) {
             CompilerDirectives.transferToInterpreter();
-            throw ArityException.create(3, arguments.length);
+            throw ArityException.create(3, 3, arguments.length);
         }
         Object value = parentInterop.execute(parent, arguments);
         try {
@@ -563,7 +563,7 @@ Object execute(Object[] arguments,
                     @CachedLibrary("this.function") InteropLibrary mapInterop) throws UnsupportedTypeException, ArityException, UnsupportedMessageException {
         if (arguments.length != 3) {
             CompilerDirectives.transferToInterpreter();
-            throw ArityException.create(3, arguments.length);
+            throw ArityException.create(3, 3, arguments.length);
         }
         Object value = parentInterop.execute(parent, arguments);
         try {
@@ -639,7 +639,7 @@ Object execute(Object[] arguments,
                     @CachedLibrary("this.parent") InteropLibrary parentInterop) throws UnsupportedTypeException, ArityException, UnsupportedMessageException {
         if (arguments.length != 3) {
             CompilerDirectives.transferToInterpreter();
-            throw ArityException.create(3, arguments.length);
+            throw ArityException.create(3, 3, arguments.length);
         }
         return new ShreddedObject(parentInterop.execute(parent, arguments));
     }
diff --git a/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/runtime/CUDARuntime.java b/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/runtime/CUDARuntime.java
@@ -900,7 +900,7 @@ public Object call(CUDARuntime cudaRuntime, Object[] args) throws ArityException
                     }
                 } else {
                     CompilerDirectives.transferToInterpreter();
-                    throw ArityException.create(3, args.length);
+                    throw ArityException.create(1, 3, args.length);
                 }
 
                 // Extract pointers;
@@ -944,7 +944,7 @@ public Object call(CUDARuntime cudaRuntime, Object[] args) throws ArityException
                     streamObj = args[3];
                 } else {
                     CompilerDirectives.transferToInterpreter();
-                    throw ArityException.create(4, args.length);
+                    throw ArityException.create(3, 4, args.length);
                 }
 
                 if (args[1] instanceof Long) {
@@ -1005,12 +1005,10 @@ public Object call(CUDARuntime cudaRuntime, Object[] args) throws ArityException
             public Object call(CUDARuntime cudaRuntime, Object[] args) throws ArityException, UnsupportedTypeException, InteropException {
                 checkArgumentLength(args, 3);
 
-                Object pointerFloat = args[0];
                 Object pointerStartEvent = args[1];
                 Object pointerEndEvent = args[2];
                 long addrStart;
                 long addrEnd;
-                long addrFloat;
 
                 if (pointerStartEvent instanceof CUDAEvent) {
                     addrStart = ((CUDAEvent) pointerStartEvent).getRawPointer();
@@ -1022,12 +1020,10 @@ public Object call(CUDARuntime cudaRuntime, Object[] args) throws ArityException
                 } else {
                     throw new GrCUDAException("expected CUDAEvent object");
                 }
-
                 try (UnsafeHelper.Float32Object floatPointer = UnsafeHelper.createFloat32Object()) {
                     callSymbol(cudaRuntime, floatPointer.getAddress(), addrStart, addrEnd );
                     return NoneValue.get();
                 }
-
             }
         },
         CUDA_EVENTRECORD("cudaEventRecord", "(pointer, pointer): sint32") {
diff --git a/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/runtime/ConfiguredKernel.java b/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/runtime/ConfiguredKernel.java
@@ -88,7 +88,7 @@ KernelArguments createKernelArguments(Object[] args, InteropLibrary booleanAcces
             throws UnsupportedTypeException, ArityException {
         if (args.length != kernel.getKernelParameters().length) {
             CompilerDirectives.transferToInterpreter();
-            throw ArityException.create(kernel.getKernelParameters().length, args.length);
+            throw ArityException.create(kernel.getKernelParameters().length, kernel.getKernelParameters().length, args.length);
         }
         KernelArguments kernelArgs = new KernelArguments(args, this.kernel.getKernelParameters());
         for (int paramIdx = 0; paramIdx < kernel.getKernelParameters().length; paramIdx++) {
diff --git a/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/runtime/Device.java b/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/runtime/Device.java
@@ -176,7 +176,9 @@ boolean isExecutable() {
     public Object execute(Object[] arguments) throws ArityException {
         if (arguments.length != 0) {
             CompilerDirectives.transferToInterpreter();
-            throw ArityException.create(0, arguments.length);
+            // FIXME: the maximum number of arguments is unbound (as each argument is a dimension of a N-dimensional tensor).
+            //  Truffle currently uses -1 to handle an unbound number of arguments;
+            throw ArityException.create(0, -1, arguments.length);
         }
         return runtime.cudaGetDevice() == deviceId;
     }
@@ -203,7 +205,7 @@ boolean isExecutable() {
     public Object execute(Object[] arguments) throws ArityException {
         if (arguments.length != 0) {
             CompilerDirectives.transferToInterpreter();
-            throw ArityException.create(0, arguments.length);
+            throw ArityException.create(0, 0, arguments.length);
         }
         runtime.cudaSetDevice(deviceId);
         return NoneValue.get();
diff --git a/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/runtime/Kernel.java b/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/runtime/Kernel.java
@@ -281,11 +281,9 @@ Object execute(Object[] arguments,
                     @CachedLibrary(limit = "3") InteropLibrary blockSizeAccess,
                     @CachedLibrary(limit = "3") InteropLibrary blockSizeElementAccess,
                     @CachedLibrary(limit = "3") InteropLibrary sharedMemoryAccess) throws UnsupportedTypeException, ArityException {
-        // FIXME: ArityException allows to specify only 1 arity, and cannot be subclassed! We might
-        // want to use a custom exception here;
         if (arguments.length < 2 || arguments.length > 4) {
             CompilerDirectives.transferToInterpreter();
-            throw ArityException.create(2, arguments.length);
+            throw ArityException.create(2, 4, arguments.length);
         }
 
         Dim3 gridSize = extractDim3(arguments[0], "gridSize", gridSizeAccess, gridSizeElementAccess);
diff --git a/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/runtime/array/AbstractArray.java b/projects/com.nvidia.grcuda/src/com/nvidia/grcuda/runtime/array/AbstractArray.java
@@ -406,7 +406,7 @@ boolean isExecutable() {
         Object execute(Object[] arguments) throws ArityException {
             if (arguments.length != 0) {
                 CompilerDirectives.transferToInterpreter();
-                throw ArityException.create(0, arguments.length);
+                throw ArityException.create(0, 0, arguments.length);
             }
             freeMemory();
             return NoneValue.get();

Original file line number	Diff line number	Diff line change
`@@ -102,7 +102,7 @@ Object execute(Object[] arguments,`
`102`	`102`	`numElements = extractNumber(arguments[1], numElementsAccess);`
`103`	`103`	`} else {`
`104`	`104`	`CompilerDirectives.transferToInterpreter();`
`105`		`- throw ArityException.create(1, arguments.length);`
	`105`	`+ throw ArityException.create(1, 2, arguments.length);`
`106`	`106`	`}`
`107`	`107`	`// Obtain what kind of copy (pointer or array) should be executed.`
`108`	`108`	`// By default, see if we can use the fast CUDA memcpy: we cannot use it if the source and target arrays`
Original file line number	Diff line number	Diff line change
`@@ -77,7 +77,7 @@ public DeviceArrayFunction(AbstractGrCUDAExecutionContext grCUDAExecutionContext`
`77`	`77`	`@TruffleBoundary`
`78`	`78`	`public Object call(Object[] arguments) throws ArityException, UnsupportedTypeException {`
`79`	`79`	`if (arguments.length < 1) {`
`80`		`- throw ArityException.create(1, arguments.length);`
	`80`	`+ throw ArityException.create(1, 2, arguments.length);`
`81`	`81`	`}`
`82`	`82`	`String typeName = expectString(arguments[0], "first argument of DeviceArray must be string (type name)");`
`83`	`83`	`Type elementType;`
Original file line number	Diff line number	Diff line change
`@@ -227,7 +227,7 @@ public Object execute(Object[] arguments,`
`227`	`227`	`@Cached MapArrayNode mapNode) throws ArityException, UnsupportedTypeException {`
`228`	`228`	`if (arguments.length < 1) {`
`229`	`229`	`CompilerDirectives.transferToInterpreter();`
`230`		`- throw ArityException.create(1, arguments.length);`
	`230`	`+ throw ArityException.create(1, 2, arguments.length);`
`231`	`231`	`}`
`232`	`232`	`String typeName;`
`233`	`233`	`try {`
`@@ -244,12 +244,11 @@ public Object execute(Object[] arguments,`
`244`	`244`	`}`
`245`	`245`	`if (arguments.length == 1) {`
`246`	`246`	`return new TypedMapDeviceArrayFunction(grCUDAExecutionContext, elementType);`
`247`		`- } else {`
`248`		`- if (arguments.length != 2) {`
`249`		`- CompilerDirectives.transferToInterpreter();`
`250`		`- throw ArityException.create(2, arguments.length);`
`251`		`- }`
	`247`	`+ } else if (arguments.length == 2) {`
`252`	`248`	`return mapNode.execute(arguments[1], elementType, grCUDAExecutionContext);`
	`249`	`+ } else {`
	`250`	`+ CompilerDirectives.transferToInterpreter();`
	`251`	`+ throw ArityException.create(1, 2, arguments.length);`
`253`	`252`	`}`
`254`	`253`	`}`
`255`	`254`	`}`
Original file line number	Diff line number	Diff line change
`@@ -61,7 +61,9 @@ public TypedDeviceArrayFunction(AbstractGrCUDAExecutionContext grCUDAExecutionCo`
`61`	`61`	`public Object call(Object[] arguments) throws ArityException, UnsupportedTypeException {`
`62`	`62`	`if (arguments.length < 1) {`
`63`	`63`	`CompilerDirectives.transferToInterpreter();`
`64`		`- throw ArityException.create(1, arguments.length);`
	`64`	`+ // FIXME: the maximum number of arguments is unbound (as each argument is a dimension of a N-dimensional tensor).`
	`65`	`+ // Truffle currently uses -1 to handle an unbound number of arguments;`
	`66`	`+ throw ArityException.create(1, -1, arguments.length);`
`65`	`67`	`}`
`66`	`68`	`return DeviceArrayFunction.createArray(arguments, 0, elementType, grCUDAExecutionContext);`
`67`	`69`	`}`
Original file line number	Diff line number	Diff line change
`@@ -64,7 +64,7 @@ public Object execute(Object[] arguments,`
`64`	`64`	`@Cached MapArrayNode mapNode) throws ArityException {`
`65`	`65`	`if (arguments.length != 1) {`
`66`	`66`	`CompilerDirectives.transferToInterpreter();`
`67`		`- throw ArityException.create(1, arguments.length);`
	`67`	`+ throw ArityException.create(1, 1, arguments.length);`
`68`	`68`	`}`
`69`	`69`	`return mapNode.execute(arguments[0], elementType, grCUDAExecutionContext);`
`70`	`70`	`}`