예제 #1
0
 public CUfunction getKernelFunction(Kernel f) {
   CUfunction function = kernels.get(f);
   if (function == null) throw new CudaException("No such function " + f);
   return function;
 }
예제 #2
0
 private void initFunction(CUmodule module, Kernel name) {
   CUfunction function = new CUfunction();
   JCudaDriver.cuModuleGetFunction(function, module, name.name());
   kernels.put(name, function);
 }