IFRT tracking issue
Opened this issue · 1 comments
mofeing commented
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::Data
andIfrtComputationClient::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