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

Enable GPU execution #88

Merged
merged 7 commits into from
Jun 24, 2020

Conversation

saiprashanth173
Copy link
Contributor

  • Add a mechanism to enable GPU execution
  • Implement a Pytorch UDF abstract class that can be used by all Pytorch based UDFs. This UDF abstract class will take care of moving models and Data to GPU.


def to_device(self, device: str):
"""

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Please add the doc string.

Copy link
Member

@gaurav274 gaurav274 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, add the missing docstring and then we can merge.
Does the existing FasterRCNN test-case work on the cluster?

@jarulraj jarulraj requested a review from xzdandy June 23, 2020 17:31
@gaurav274 gaurav274 merged commit 02345f0 into georgia-tech-db:master Jun 24, 2020
llxxll12345 pushed a commit that referenced this pull request Feb 15, 2021
luoj1 pushed a commit to luoj1/eva that referenced this pull request Apr 29, 2023
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.

2 participants