Lines Matching refs:A_d

94   T** A_d;  in getrfBatched()  local
99 arch::malloc((void**)&A_d, batchSize * sizeof(*A_h)); in getrfBatched()
100 arch::memcopy(A_d, A_h, batchSize * sizeof(*A_h), arch::memcopyH2D); in getrfBatched()
101 …cublasStatus_t status = cublas::cublas_getrfBatched(*(a[0]).handles.cublas_handle, n, A_d, lda, to… in getrfBatched()
105 arch::free(A_d); in getrfBatched()
156 T **A_d, **C_d; in getriBatched() local
165 arch::malloc((void**)&A_d, batchSize * sizeof(*A_h)); in getriBatched()
167 arch::memcopy(A_d, A_h, batchSize * sizeof(*A_h), arch::memcopyH2D); in getriBatched()
169 …cublasStatus_t status = cublas::cublas_getriBatched(*(a[0]).handles.cublas_handle, n, A_d, lda, to… in getriBatched()
173 arch::free(A_d); in getriBatched()
189 T1 **A_d, **A_h; in matinvBatched() local
198 arch::malloc((void**)&A_d, batchSize * sizeof(*A_h)); in matinvBatched()
200 arch::memcopy(A_d, A_h, batchSize * sizeof(*A_h), arch::memcopyH2D); in matinvBatched()
202 …cublasStatus_t status = cublas::cublas_matinvBatched(*(a[0]).handles.cublas_handle, n, A_d, lda, C… in matinvBatched()
206 arch::free(A_d); in matinvBatched()
372 T** A_d(B_d); in geqrfBatched() local
374 …cublasStatus_t status = cublas::cublas_geqrfBatched(*(A[0]).handles.cublas_handle, M, N, A_d, LDA,… in geqrfBatched()
415 T **A_d, **T_d; in geqrfStrided() local
416 arch::malloc((void**)&A_d, batchSize * sizeof(*A_h)); in geqrfStrided()
417 arch::memcopy(A_d, A_h, batchSize * sizeof(*A_h), arch::memcopyH2D); in geqrfStrided()
421 …cublas::cublas_geqrfBatched(*A.handles.cublas_handle, M, N, A_d, LDA, T_d, to_address(inf.data()),… in geqrfStrided()
426 arch::free(A_d); in geqrfStrided()