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.

Sign up for GitHub

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

Jump to bottom

How to control the number of threads? #63

Open

avik-pal opened this issue Aug 11, 2024 · 2 comments

Collaborator

avik-pal commented Aug 11, 2024

Setting up some benchmarking, is there a way to control the number of threads being used? I am guessing the julia threads count doesn't propagate.

Collaborator Author

avik-pal commented Aug 13, 2024

Will still need to confirm if this works. But jax-ml/jax#743 is probably the way to go.

Collaborator

mofeing commented Aug 13, 2024

Great! I was questioning the same thing. Can't it be controlled through PJRT API?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment