异步cuBLAS调用

11

我想异步调用cuBLAS例程。这是否可能?如果是,我该如何实现?

1个回答

11

在进行cublas调用之前,请使用cublasSetStream函数。

cublasSetStream(cublasHandle,cudaStream);


但是我在我的gemv和gemm调用中没有使用这个句柄,那么它还能工作吗? - user1439690
1
在新的cublas api中,此函数已更改为cublasSetKernelStream(cudaStream) - sgarizvi
从文档上看不清楚是否只需要一个非默认流,还是每个异步计算都应该使用一个流。 - Michael Conlen
1
该函数现在变为简单的 cublasSetStream(handle, streamId) - una_dinosauria

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接