apache / tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators
https://tvm.apache.org/
Apache License 2.0
11.41k stars 3.4k forks source link

[RPC] Raise error if server process terminated #17101

Closed Lunderberg closed 2 weeks ago

Lunderberg commented 2 weeks ago

Prior to this PR, a local RPC server could crash without any indication in the main process. While typically this crash would cause an error in the main process due to the lack of a RPCCode::kReturn from the server, the delayed error can complicate debugging.

This PR updates the local RPC server to raise an exception if the server process returns with a non-zero exit code.