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

New inf2 runtime support with nos serve up #535

Merged
merged 1 commit into from
Jan 30, 2024

Conversation

spillai
Copy link
Contributor

@spillai spillai commented Jan 30, 2024

Summary

This allows the user to run nos serve up -c <serve.yaml> and automatically run on inf2 environments

  • added inf2 device type in serve.yml.j2 template
  • added devices field to programmatically determine number of neuron devices at startup

Related issues

Checks

  • make lint: I've run make lint to lint the changes in this PR.
  • make test: I've made sure the tests (make test-cpu or make test) are passing.
  • Additional tests:
    • Benchmark tests (when contributing new models)
    • GPU/HW tests

@spillai spillai added the distribution Distribution related issues label Jan 30, 2024
@spillai spillai added this to the NOS v0.2 milestone Jan 30, 2024
@spillai spillai requested a review from outtanames January 30, 2024 07:06
@spillai spillai self-assigned this Jan 30, 2024
 - added `inf2` device type in `serve.yml.j2` template
 - added `devices` field to programmatically determine number of
 neuron devices at startup
@spillai spillai merged commit bd1d2c1 into autonomi-ai:main Jan 30, 2024
2 checks passed
@spillai spillai deleted the spillai/0.2.0-dev branch January 30, 2024 21:59
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
distribution Distribution related issues
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant