Java通过JNI调用CUDA程序

最新推荐文章于 2024-07-11 13:26:17 发布

南蛮小子

最新推荐文章于 2024-07-11 13:26:17 发布

阅读量7.8k

点赞数 3

分类专栏：编程语言文章标签： java cuda jni GPU nsight

本文链接：https://blog.csdn.net/lllzzh123321/article/details/53520815

版权

本文介绍了如何使用Java的JNI接口调用CUDA程序，包括编写Java代码声明调用接口，生成调用头文件，编写CUDA程序，编译生成动态链接库，以及在Java中加载并调用CUDA程序的详细步骤。特别提到了使用Nsight进行CUDA程序开发的便利性，以及动态链接库的加载和使用注意事项。

摘要由CSDN通过智能技术生成

在这里不讨论为什么不使用Jcuda，只是说明如何通过JNI（Java Native Interface）调用CUDA程序
（1）编写java程序通过native关键字声明调用接口
（2）生成调用头文件xx.h，使用javah命令生成（如果你够牛逼可以自己手写，但是这样容易出错）
（3）编写CUDA程序（跟编写C/C++差不多）
（4）编译生成动态链接库libxx.so文件（windows下应该是xx.dll文件，我这里讲的都是基于linux下的）
（5）java加载上述动态链接库，然后就可以通过接口调用CUDA程序了
（6）编译运行，OK

说的挺简单了，其实有很多细节要注意的

（1）编写java程序通过native关键字声明调用接口

package com.lzh
public class Hello
{
   
    static 
    {
        try
        {
            System.loadLibrary("sharedLibraryProject");
        }
        catch(UnsatisfiedLinkError e)
        {
            System.err.println("can not load hello library :\n"+
            e.toString());
        }

    }
    public native void SayHello(String strName);
    public native int Add(int a,int b);

}