2012-09-25 91 views

回答

8

在cublas調用之前使用cublasSetStream函數。

cublasSetStream(cublasHandle, cudaStream);

+0

但我沒有在我的任何gemv和gemm調用中使用句柄,所以它會工作嗎? – user1439690

+1

在新的cublas api中,這個函數被改爲'cublasSetKernelStream(cudaStream)' – sgarizvi

+0

如果你只需要一個非默認流或者如果你想爲異步發生的每個計算使用一個流,就不清楚看文檔。 –