elixir-nx / nx

Multi-dimensional arrays (tensors) and numerical definitions for Elixir
2.66k stars 194 forks source link

feat: add to/from pointer #1473

Closed polvalente closed 7 months ago

polvalente commented 7 months ago

Enables pointer manipulation for sharing CUDA memory between processes in EXLA. In theory, IPC can also be supported for host-allocated memory, but this is not implemented yet. It is also possible to use local (as in, "same process") pointers through these functions, albeit there's not much gain besides automated testing.