Skip to content

Conversation

@tongzhou80
Copy link
Contributor

@tongzhou80 tongzhou80 commented Dec 24, 2024

Hi Alex, Could you review when you get a chance? Thanks a lot! @alexnick83

For now I've just added the APPy code for one benchmark. Will try to add more if the PR works well.

Copy link
Contributor

@alexnick83 alexnick83 left a comment

Choose a reason for hiding this comment

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

Looks good. I see that you are using pytorch to get the data on the GPU. This is not an issue since there is interoperability between torch and cupy, but we may need to update how data are validated.

@tongzhou80
Copy link
Contributor Author

tongzhou80 commented Feb 3, 2025

Looks good. I see that you are using pytorch to get the data on the GPU. This is not an issue since there is interoperability between torch and cupy, but we may need to update how data are validated.

Ah I see, thanks for taking a look and pointing that out. I'll fix that and send another PR, either by just using CuPy arrays as input/output, or updating the validation code to also allow pytorch.

@alexnick83 alexnick83 merged commit 35cb9f3 into spcl:main Feb 5, 2025
1 check passed
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