Lines Matching refs:PetscCallCUPM
283 …PetscCallCUPM(cupmMemcpyAsync(buf, (char *)data + start * link->unitbytes, count * link->unitbytes… in Pack()
299 …PetscCallCUPM(cupmMemcpyAsync((char *)data + start * link->unitbytes, buf, count * link->unitbytes… in UnpackAndOp()
368 …PetscCallCUPM(cupmMemcpyAsync((Type *)dst + dstStart * link->bs, (const Type *)src + srcStart * li… in ScatterAndInsert()
569 PetscCallCUPM(cupmDeviceSynchronize()); in LinkSyncDevice()
577 PetscCallCUPM(cupmStreamSynchronize(link->stream)); in LinkSyncStream()
596 PetscCallCUPM(cupmMemcpyAsync(dst, src, n, kinds[stype][dtype], link->stream)); in LinkMemcpy()
609 PetscCallCUPM(cupmMalloc(ptr, size)); in Malloc()
619 else if (PetscMemTypeDevice(mtype)) PetscCallCUPM(cupmFree(ptr)); in Free()
630 PetscCallCUPM(cupmFree(link->rootbuf_alloc[i][PETSC_MEMTYPE_DEVICE])); in LinkDestroy_MPI()
631 PetscCallCUPM(cupmFree(link->leafbuf_alloc[i][PETSC_MEMTYPE_DEVICE])); in LinkDestroy_MPI()
777 PetscCallCUPM(cupmGetDevice(&device)); in LinkSetUp()
778 PetscCallCUPM(cupmGetDeviceProperties(&props, device)); in LinkSetUp()