Skip to content

Pass device memory to MParT through bindings #399

@mparno

Description

@mparno

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.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    Status

    In Progress

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions