-
Notifications
You must be signed in to change notification settings - Fork 38
Description
Based on https://github.com/pytorch/xla/blob/master/torch_xla/csrc/runtime/ifrt_computation_client.h, we might be interested in implementing some functions similar to the following C++ methods in ReactantExtra:
- From
IfrtComputationClient-
CreateDataPlaceholder -
GetDataShards,GetDataShard -
not implementedGetPjRtBuffer -
Compile -
ExecuteComputation -
GetNumDevices -
GetDefaultDevice -
GetNumProcesses
-
- Some functionality similar to
ComputationClient::DataandIfrtComputationClient::IfrtData?
Also it might be interesting to get some ideas for abstracting both PjRt and IFRT structures into one common interface by looking to https://github.com/pytorch/xla/blob/master/torch_xla/csrc/runtime/computation_client.h
Steps
- Get some IFRT buffer object and move data in and out of it
- Compile sth with this buffer