Class JCudaKernels
- java.lang.Object
-
- org.apache.sysds.runtime.instructions.gpu.context.JCudaKernels
-
public class JCudaKernels extends Object
-
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description static void
checkResult(int cuResult)
void
launchKernel(String name, ExecutionConfig config, Object... arguments)
Setups the kernel parameters and launches the kernel using cuLaunchKernel API.
-
-
-
Method Detail
-
launchKernel
public void launchKernel(String name, ExecutionConfig config, Object... arguments)
Setups the kernel parameters and launches the kernel using cuLaunchKernel API. This function currently supports two dimensional grid and blocks.- Parameters:
name
- name of the kernelconfig
- execution configurationarguments
- can be of type Pointer, long, double, float and int
-
checkResult
public static void checkResult(int cuResult)
-
-