2012-09-25 16 views

Odpowiedz

8

Użyj funkcji cublasSetStream przed wywołaniem Cublas.

cublasSetStream(cublasHandle, cudaStream);

+0

Ale nie używam klamki w żadnym z moich połączeń gemv i gemm, więc czy to też działa? – user1439690

+1

W nowej aplikacji CPI ta funkcja została zmieniona na 'cublasSetKernelStream (cudaStream)' – sgarizvi

+0

Nie jest jasne, patrząc na dokumentację, jeśli potrzebujesz tylko pojedynczego strumienia domyślnego lub jeśli powinieneś użyć strumienia dla każdego obliczenia, które chcesz przeprowadzić asynchronicznie. –