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

Add a PyTorch worker #211

Open
ZchiPitt opened this issue Jul 8, 2023 · 0 comments · May be fixed by #212
Open

Add a PyTorch worker #211

ZchiPitt opened this issue Jul 8, 2023 · 0 comments · May be fixed by #212

Comments

@ZchiPitt
Copy link
Contributor

ZchiPitt commented Jul 8, 2023

Is your feature request related to a problem? Please describe.
We need to add a PyTorch worker that accepts the name of an PyTorch model file as an argument and compiles and evaluates it.

Describe the solution you'd like

  • install pytorch and make the build succeed
  • Get a small model () to run instead of the modified resnet50 of PtZendnn
  • add a test for PyTorch worker
  • add a new stage in the Dockerfile to build the PyTorch

Describe alternatives you've considered
NA

Additional context
NA

@ZchiPitt ZchiPitt linked a pull request Jul 25, 2023 that will close this issue
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 a pull request may close this issue.

1 participant