search

NVIDIA / AMGX

Distributed multigrid linear solver library on GPU
468 stars 136 forks source link

Benchmarking #278

Open kai-lan opened 8 months ago

kai-lan commented 8 months ago

I haven't found any tutorials on benchmarking. So far I have seen setup time for AMG really low even for large matrices. I wonder whether the setup is done on CPU or GPU and how to properly time it.

mattmartineau commented 8 months ago

Setup is performed on the GPU, you should see the time taken in the setup and solve phases if you add the config flag "obtain_timings": 1 in the solver scope.

You could also use Nsight Systems to take a look at what is happening on the GPU, if it is of interest.