Searched refs:dmem (Results 1 – 2 of 2) sorted by relevance
4354 bool dmem; in MatSeqAIJCopySubArray_SeqAIJHIPSPARSE() local4358 dmem = isHipMem(v); in MatSeqAIJCopySubArray_SeqAIJHIPSPARSE()4367 if (dmem) dv = thrust::device_pointer_cast(v); in MatSeqAIJCopySubArray_SeqAIJHIPSPARSE()4379 …} else PetscCallHIP(hipMemcpy(v, av, n * sizeof(PetscScalar), dmem ? hipMemcpyDeviceToDevice : hip… in MatSeqAIJCopySubArray_SeqAIJHIPSPARSE()4381 if (!dmem) PetscCall(PetscLogCpuToGpu(n * sizeof(PetscScalar))); in MatSeqAIJCopySubArray_SeqAIJHIPSPARSE()
5035 bool dmem; in MatSeqAIJCopySubArray_SeqAIJCUSPARSE() local5039 dmem = isCudaMem(v); in MatSeqAIJCopySubArray_SeqAIJCUSPARSE()5048 if (dmem) { in MatSeqAIJCopySubArray_SeqAIJCUSPARSE()5062 …PetscCallCUDA(cudaMemcpy(v, av, n * sizeof(PetscScalar), dmem ? cudaMemcpyDeviceToDevice : cudaMem… in MatSeqAIJCopySubArray_SeqAIJCUSPARSE()5064 if (!dmem) PetscCall(PetscLogCpuToGpu(n * sizeof(PetscScalar))); in MatSeqAIJCopySubArray_SeqAIJCUSPARSE()