You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Currently, we are hardcoding the number of addressable devices to be returned when PJRT_LoadedExecutable_AddressableDevices, to be able to support multichip, this has to be custom, with this information inferred from the StableHLO code, possibly via the mhlo.num_partitions = 2 attribute. This will require changes to the ExecutableImage::get_num_addressable_devices() function. It also might necessitate changing this part of the code that returns concrete addressable devices, but is currently not tested:
Currently, we are hardcoding the number of addressable devices to be returned when
PJRT_LoadedExecutable_AddressableDevices
, to be able to support multichip, this has to be custom, with this information inferred from the StableHLO code, possibly via themhlo.num_partitions = 2
attribute. This will require changes to theExecutableImage::get_num_addressable_devices()
function. It also might necessitate changing this part of the code that returns concrete addressable devices, but is currently not tested:The text was updated successfully, but these errors were encountered: