Lines Matching refs:handle
378 cupmSolverHandle_t handle; in Factor() local
383 PetscCall(GetHandles_(&dctx, &handle, &stream)); in Factor()
396 return cupmSolverXgetrf_bufferSize(handle, m, n, da.cupmdata(), lda, fact_lwork); in Factor()
404 …PetscCallCUPMSOLVER(cupmSolverXgetrf(handle, m, n, da.cupmdata(), lda, mcu->d_fact_work, mcu->d_fa… in Factor()
418 cupmSolverHandle_t handle; in Solve() local
421 PetscCall(GetHandlesFrom_(dctx, &handle)); in Solve()
436 handle, op, m, nrhs, da.cupmdata(), lda, fact_ipiv, x, ldx, lwork in Solve()
442 …PetscCallCUPMSOLVER(cupmSolverXgetrs(handle, op, m, nrhs, da.cupmdata(), lda, fact_ipiv, x, ldx, m… in Solve()
462 cupmSolverHandle_t handle; in Factor() local
468 PetscCall(GetHandles_(&dctx, &handle, &stream)); in Factor()
482 handle, CUPMSOLVER_FILL_MODE_LOWER, n, da.cupmdata(), lda, fact_lwork in Factor()
489 …PetscCallCUPMSOLVER(cupmSolverXpotrf(handle, CUPMSOLVER_FILL_MODE_LOWER, n, da.cupmdata(), lda, mc… in Factor()
501 …PetscCallCUPMSOLVER(cupmSolverDnXsytrf_bufferSize(handle, n, da.cupmdata(), lda, &mcu->d_fact_lwor… in Factor()
506 …PetscCallCUPMSOLVER(cupmSolverXsytrf(handle, CUPMSOLVER_FILL_MODE_LOWER, n, da, lda, mcu->d_fact_i… in Factor()
517 cupmSolverHandle_t handle; in Solve() local
521 PetscCall(GetHandlesFrom_(dctx, &handle)); in Solve()
535 handle, CUPMSOLVER_FILL_MODE_LOWER, m, nrhs, da.cupmdata(), lda, x, ldx, lwork in Solve()
541 …PetscCallCUPMSOLVER(cupmSolverXpotrs(handle, CUPMSOLVER_FILL_MODE_LOWER, m, nrhs, da.cupmdata(), l… in Solve()
564 cupmSolverHandle_t handle; in Factor() local
569 PetscCall(GetHandles_(&dctx, &handle, &stream)); in Factor()
585 return cupmSolverXgeqrf_bufferSize(handle, m, n, da.cupmdata(), lda, fact_lwork); in Factor()
591 …PetscCallCUPMSOLVER(cupmSolverXgeqrf(handle, m, n, da.cupmdata(), lda, mcu->d_fact_tau, mcu->d_fac… in Factor()
811 cupmBlasHandle_t handle; in MatMultAddColumnRange_Dispatch_() local
852 PetscCall(GetHandles_(&dctx, &handle)); in MatMultAddColumnRange_Dispatch_()
862 …PetscCallCUPMBLAS(cupmBlasXgemv(handle, op, m, n, &one, da.cupmdata() + c_start * lda, lda, dxx.cu… in MatMultAddColumnRange_Dispatch_()
1300 cupmBlasHandle_t handle; in MatMatMult_Numeric_Dispatch() local
1315 PetscCall(GetHandles_(&dctx, &handle)); in MatMatMult_Numeric_Dispatch()
1329 …PetscCallCUPMBLAS(cupmBlasXgemm(handle, transpose_A ? CUPMBLAS_OP_T : CUPMBLAS_OP_N, transpose_B ?… in MatMatMult_Numeric_Dispatch()
1543 cupmBlasHandle_t handle; in Scale() local
1545 PetscCall(GetHandlesFrom_(dctx, &handle)); in Scale()
1547 PetscCallCUPMBLAS(cupmBlasXscal(handle, N, &cu_alpha, da.cupmdata(), 1)); in Scale()
1592 cupmBlasHandle_t handle; in AXPY() local
1594 PetscCall(GetHandlesFrom_(dctx, &handle)); in AXPY()
1596 PetscCallCUPMBLAS(cupmBlasXaxpy(handle, N, &cu_alpha, dx.cupmdata(), 1, dy.cupmdata(), 1)); in AXPY()
1786 cupmSolverHandle_t handle; in InvertFactors() local
1800 PetscCall(GetHandles_(&dctx, &handle, &stream)); in InvertFactors()
1806 …PetscCallCUPMSOLVER(cupmSolverXpotri_bufferSize(handle, CUPMSOLVER_FILL_MODE_LOWER, n, da.cupmdata… in InvertFactors()
1813 …PetscCallCUPMSOLVER(cupmSolverXpotri(handle, CUPMSOLVER_FILL_MODE_LOWER, n, da.cupmdata(), lda, mc… in InvertFactors()