PjRtComputationClient::ExecuteReplicated core dump when encountering a scalar

[mars1248]

❓ Questions and Help

In my test code, I found that there might be PjRtData as the type argument(the argument is a scalar), and then the core dump. https://github.com/pytorch/xla/blob/master/torch_xla/csrc/runtime/pjrt_computation_client.cc#L806 I wrote a test function earlier that tried to transform all arguments manually, but core dumped.