Lines Matching refs:offloadmask
212 if (A->offloadmask == PETSC_OFFLOAD_CPU) { // A's latest factors are on CPU in MatSeqAIJCUSPARSEBuildFactoredMatrix_LU()
319 if (A->offloadmask == PETSC_OFFLOAD_UNALLOCATED || A->offloadmask == PETSC_OFFLOAD_CPU) { in MatSeqAIJCUSPARSEBuildILULowerTriMatrix()
451 if (A->offloadmask == PETSC_OFFLOAD_UNALLOCATED || A->offloadmask == PETSC_OFFLOAD_CPU) { in MatSeqAIJCUSPARSEBuildILUUpperTriMatrix()
590 A->offloadmask = PETSC_OFFLOAD_BOTH; // factored matrix is sync'ed to GPU in MatSeqAIJCUSPARSEILUAnalysisAndCopyToGPU()
630 if (A->offloadmask == PETSC_OFFLOAD_CPU) { // A's latest factors are on CPU in MatSeqAIJCUSPARSEBuildFactoredMatrix_Cholesky()
789 if (A->offloadmask == PETSC_OFFLOAD_UNALLOCATED || A->offloadmask == PETSC_OFFLOAD_CPU) { in MatSeqAIJCUSPARSEBuildICCTriMatrices()
990 A->offloadmask = PETSC_OFFLOAD_BOTH; in MatSeqAIJCUSPARSEICCAnalysisAndCopyToGPU()
1018 B->offloadmask = PETSC_OFFLOAD_CPU; in MatCholeskyFactorNumeric_SeqAIJCUSPARSE()
1749 fact->offloadmask = PETSC_OFFLOAD_GPU; in MatILUFactorNumeric_SeqAIJCUSPARSE_ILU0()
1784 fact->offloadmask = PETSC_OFFLOAD_BOTH; in MatILUFactorSymbolic_SeqAIJCUSPARSE_ILU0()
1999 fact->offloadmask = PETSC_OFFLOAD_GPU; in MatICCFactorNumeric_SeqAIJCUSPARSE_ICC0()
2033 fact->offloadmask = PETSC_OFFLOAD_BOTH; in MatICCFactorSymbolic_SeqAIJCUSPARSE_ICC0()
2158 B->offloadmask = PETSC_OFFLOAD_CPU; in MatLUFactorNumeric_SeqAIJCUSPARSE()
2325 if (A->offloadmask == PETSC_OFFLOAD_GPU) { in MatSeqAIJCUSPARSECopyFromGPU()
2341 A->offloadmask = PETSC_OFFLOAD_BOTH; in MatSeqAIJCUSPARSECopyFromGPU()
2357 A->offloadmask = PETSC_OFFLOAD_CPU; in MatSeqAIJRestoreArray_SeqAIJCUSPARSE()
2387 A->offloadmask = PETSC_OFFLOAD_CPU; in MatSeqAIJRestoreArrayWrite_SeqAIJCUSPARSE()
2434 if (A->offloadmask == PETSC_OFFLOAD_UNALLOCATED || A->offloadmask == PETSC_OFFLOAD_CPU) { in MatSeqAIJCUSPARSECopyToGPU()
2567 if (both) A->offloadmask = PETSC_OFFLOAD_BOTH; in MatSeqAIJCUSPARSECopyToGPU()
3010 C->offloadmask = PETSC_OFFLOAD_GPU; in MatProductNumeric_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3381 C->offloadmask = PETSC_OFFLOAD_UNALLOCATED; in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3385 …if (product->api_user && A->offloadmask == PETSC_OFFLOAD_BOTH && B->offloadmask == PETSC_OFFLOAD_B… in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3387 C->offloadmask = PETSC_OFFLOAD_GPU; in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3762 if (A->offloadmask == PETSC_OFFLOAD_BOTH || A->offloadmask == PETSC_OFFLOAD_GPU) { in MatGetDiagonal_SeqAIJCUSPARSE()
3989 if (gpu) A->offloadmask = PETSC_OFFLOAD_GPU; in MatZeroEntries_SeqAIJCUSPARSE()
3992 A->offloadmask = PETSC_OFFLOAD_CPU; in MatZeroEntries_SeqAIJCUSPARSE()
4113 B->offloadmask = PETSC_OFFLOAD_UNALLOCATED; in MatConvert_SeqAIJ_SeqAIJCUSPARSE()
4394 mat->offloadmask = PETSC_OFFLOAD_CPU; in MatSetPreallocationCOO_SeqAIJCUSPARSE()
4630 A->offloadmask = PETSC_OFFLOAD_GPU; in MatSeqAIJCUSPARSEGetArray()
4692 A->offloadmask = PETSC_OFFLOAD_GPU; in MatSeqAIJCUSPARSEGetArrayWrite()
5029 (*C)->offloadmask = PETSC_OFFLOAD_GPU; in MatSeqAIJCUSPARSEMergeMats()