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.

Sign up for GitHub

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

Jump to bottom

Add dlpack support #746

Open

SunDoge wants to merge 3 commits into LaurentMazare:main from SunDoge:feat-dlpack

Contributor

SunDoge commented Jun 16, 2023

DLPack is an open in-memory tensor structure for sharing tensors among frameworks. DLPack enables

Easier sharing of operators between deep learning frameworks.
Easier wrapping of vendor level operator implementations, allowing collaboration when introducing new devices/ops.
Quick swapping of backend implementations, like different version of BLAS
For final users, this could bring more operators, and possibility of mixing usage between frameworks.

Supporting dlpack will make it easier to share tensors with dfdx, ndarray and maybe OrtTensor.

SunDoge added 2 commits

June 15, 2023 19:23


          add to_dlpack and from_dlpack

c9a8e95


          from vec zero copy

fd37de7

SunDoge mentioned this pull request

Conversion to/from tch Tensors pykeio/ort#46

Closed


          update api

cc23806

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment