Skip to content

Conversation

@aws-taylor
Copy link

Before or while filing an issue please feel free to join our slack channel to get in touch with development team, ask questions, find out what's cooking and more!

Issue #, if available:

Description of changes:

Similar to GPUs, when running on AWS Inf1 instance types, automatically detect the number of available Neuron cores and set the number of workers accordingly.

Testing done:

Ran tests on inf1, logs correctly show 16 neuron cores on an inf1.6xlarge.

...
   Temp directory: /tmp
    Number of GPUs: 0
    Number of Neuron Cores: 16
    Number of CPUs: 24
    Max heap size: 10457 M
    Python executable: python
...

To run CI tests on your changes refer README.md

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.

@maaquib maaquib requested a review from lxning August 24, 2022 18:51
@aws-taylor
Copy link
Author

aws-taylor commented Aug 24, 2022 via email

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

Successfully merging this pull request may close these issues.

3 participants