Commit 220fa040 by Shuai Yuan Committed by Tianqi Chen

[RPC] clarify error message for unmatched context (#451)

Clarify confusing error message for unmatched context
parent ffff1e49
...@@ -149,7 +149,7 @@ class RPCSession::EventHandler { ...@@ -149,7 +149,7 @@ class RPCSession::EventHandler {
TVMContext StripSessMask(TVMContext ctx) { TVMContext StripSessMask(TVMContext ctx) {
int dev_type = ctx.device_type; int dev_type = ctx.device_type;
CHECK_EQ(dev_type / kRPCSessMask, rpc_sess_table_index_ + 1) CHECK_EQ(dev_type / kRPCSessMask, rpc_sess_table_index_ + 1)
<< "Can only TVMContext related to the same remote sesstion"; << "Can not pass in local context or context with a different remote session";
ctx.device_type = static_cast<DLDeviceType>(dev_type % kRPCSessMask); ctx.device_type = static_cast<DLDeviceType>(dev_type % kRPCSessMask);
return ctx; return ctx;
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment