Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Ability to select GPU? #28

Open
qingshanchen opened this issue Jun 30, 2021 · 0 comments
Open

Ability to select GPU? #28

qingshanchen opened this issue Jun 30, 2021 · 0 comments

Comments

@qingshanchen
Copy link

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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant