asynchronous - 异步 cuBLAS 调用

Question

我想异步调用 cuBLAS 例程。是否可以？如果是，我该如何实现？

Answer 1

cublasSetStream在 cublas 调用之前使用该函数。

cublasSetStream(cublasHandle, cudaStream);