Commit Graph

1 Commits

Author SHA1 Message Date
Peter Hawkins
fc1f6fdf94 [XLA:Python] Add DLPack import/export support to the XLA Python client.
This allows JAX to communicate on-device arrays with other libraries, such as PyTorch and CuPy.

PiperOrigin-RevId: 290845329
Change-Id: Idd99d81533159bc2ad0c5177b69ac7f30315cb1a
2020-01-21 16:15:21 -08:00