[petsc-dev] PetscMallocAlign for Cuda

Barry Smith bsmith at petsc.dev
Wed Sep 2 12:33:39 CDT 2020


  Mark,

   Currently you use directly the Nvidia provided mallocs cudaMalloc for all mallocs on the GPU. See for example aijcusparse.cu. 

   I will be using Stefano's work to start developing a unified PETSc based system for all memory management but don't wait for that.

   Barry


> On Sep 2, 2020, at 8:58 AM, Mark Adams <mfadams at lbl.gov> wrote:
> 
> PETSc mallocs seem to boil down to PetscMallocAlign. There are switches in here but I don't see a Cuda malloc. THis would seem to be convenient if I want to create an Object entirely on Cuda or any device. 
> 
> Are there any thoughts along these lines or should I just duplicate Mat creation, for instance, by hand?



More information about the petsc-dev mailing list