pyamgx icon indicating copy to clipboard operation
pyamgx copied to clipboard

Ability to select GPU?

Open qingshanchen opened this issue 3 years ago • 0 comments

I understand that currently pyamgx only supports "simple" Resource objects for single GPU application. However, in a multi-GPU environment, does pyamgx currently have the ability to choose which GPU to use? It does not seem so, and the following is from the current doc page: "Resources objects are used to specify the resources (GPUs, MPI ranks) used by Vector, Matrix and Solver objects. Currently, pyamgx only supports “simple” Resources objects for single threaded, single GPU applications. created using the create_simple() method:"

It appears that the AMGX library only supports single-GPU applications. But the ability to choose which GPU to run on can still be beneficial, because otherwise the application will always runson the default (#0) GPU, which is often the busiest one.

qingshanchen avatar Jun 30 '21 21:06 qingshanchen