Skip to content
This repository has been archived by the owner on Oct 31, 2023. It is now read-only.

Add support for PyTorch in LoopTool #11

Open
bwasti opened this issue May 17, 2022 · 0 comments
Open

Add support for PyTorch in LoopTool #11

bwasti opened this issue May 17, 2022 · 0 comments

Comments

@bwasti
Copy link
Contributor

bwasti commented May 17, 2022

Currently, NumPy is integrated through py::array_t (a pybind11 builtin). This allows users to create LoopTool native tensors from NumpPy NDArrays.

https://github.com/facebookresearch/loop_tool/blob/main/src/frontends/python.cpp#L51-L69

Notably, there is an optional "copy" flag that allows the user to instead the underlying NumPy data, saving on reads/writes.

Both of these features can be adopted for PyTorch Tensors.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant