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

Pass device memory to MParT through bindings #399

Open
mparno opened this issue Mar 26, 2024 · 1 comment
Open

Pass device memory to MParT through bindings #399

mparno opened this issue Mar 26, 2024 · 1 comment
Milestone

Comments

@mparno
Copy link
Contributor

mparno commented Mar 26, 2024

Ultimately I'd like to create a pytorch Tensor (or Julia array), send it to the device, and then wrap that GPU-memory in a way that Kokkos can use inside MParT. This issue is devoted to assessing the feasibility of that and possibly an initial implementation in the python bindings.

@mparno mparno added this to the v3 milestone Mar 26, 2024
@dannys4
Copy link
Contributor

dannys4 commented Mar 26, 2024

To build on this/make a suggestion, I'd like to just have an agnostic wrapper for a Kokkos::View that's like MParTArray or something, which we can (in the bindings) create a toBinding or something to create the appropriate array in Python/Julia, as well as toHost, so that passing this in allows us to evaluate maps in the bindings without copying to or from the device

@dannys4 dannys4 modified the milestones: v3, v3.1 May 16, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
Development

No branches or pull requests

2 participants