2012-09-25 2 views

Répondre

8

Utilisez la fonction cublasSetStream avant les appels cublas.

cublasSetStream(cublasHandle, cudaStream);

+0

Mais je n'utilise pas le handle dans aucun de mes appels gemv et gemm alors cela fonctionnera-t-il aussi? – user1439690

+1

Dans la nouvelle API de cublas, cette fonction a été changée en 'cublasSetKernelStream (cudaStream)' – sgarizvi

+0

Il n'est pas évident de regarder la documentation si vous avez besoin d'un seul flux par défaut ou si vous devez utiliser un flux pour chaque calcul asynchrone. –