如何解决为什么在 Cusolver 中每隔一个函数调用才计算特征向量?
我正在编写一个程序来比较 JBlas 和 JCublas 的速度。当我第一次调用下面的函数时,一切正常,并且 v 包含正确的特征向量。当我第二次调用它时,它的计算时间要少得多,但只返回输入的对称矩阵 a
,就好像 d_A
的值从未改变过一样。
该函数似乎只能在奇数调用上按预期工作。我有一种预感,这个错误是由于 GPU 内存中的某些东西没有被正确清除,但我找不到它。
public static void getSymEigenGPU(cusolverDnHandle handle,DoubleMatrix a) {
int n2 = a.length;
int n = a.rows;
double[] a1d = to1d(a);
double[] v = new double[n2];
double[] w = new double[n];
Pointer h_A = Pointer.to(a1d);
Pointer h_V = Pointer.to(v);
Pointer h_W = Pointer.to(w);
Pointer d_A = new Pointer();
Pointer d_V = new Pointer();
Pointer d_W = new Pointer();
Pointer d_work = new Pointer();
JCuda.cudamalloc(d_A,(long) n2 * Sizeof.DOUBLE);
JCuda.cudamalloc(d_V,(long) n2 * Sizeof.DOUBLE);
JCuda.cudamalloc(d_W,n * Sizeof.DOUBLE);
int jobz = CUSOLVER_EIG_MODE_VECTOR;
int uplo = CUBLAS_FILL_MODE_UPPER;
JCuda.cudamemcpy(d_A,h_A,(long) n2 * Sizeof.DOUBLE,cudamemcpyHostToDevice);
int[] lworkl = new int[1];
JCusolverDn.cusolverDnDsyevd_bufferSize(handle,jobz,uplo,n,d_A,d_W,lworkl);
int lwork = lworkl[0];
JCuda.cudamalloc(d_work,(long) lwork * Sizeof.DOUBLE);
NanoStopWatch sw = NanoStopWatch.sw();
JCusolverDn.cusolverDnDsyevd(handle,d_work,n2,new Pointer());
System.out.println("sw.stop() = " + sw.stop());
JCuda.cudamemcpy(h_W,Sizeof.DOUBLE * n,cudamemcpyDevicetoHost);
JCuda.cudamemcpy(h_V,(long) Sizeof.DOUBLE * n2,cudamemcpyDevicetoHost);
pp(from1d(v));
JCuda.cudaFree(d_A);
JCuda.cudaFree(d_V);
JCuda.cudaFree(d_W);
JCuda.cudaFree(d_work);
}
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。