From 220fa040152c60f85251d5b72dbd09ffee3f4ad4 Mon Sep 17 00:00:00 2001
From: Shuai Yuan <ysh329@users.noreply.github.com>
Date: Tue, 12 Sep 2017 11:24:42 -0500
Subject: [PATCH] [RPC] clarify error message for unmatched context (#451)
Clarify confusing error message for unmatched context
---
src/runtime/rpc/rpc_session.cc | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/runtime/rpc/rpc_session.cc b/src/runtime/rpc/rpc_session.cc
index 7bdb4d3ef..323faf4a9 100644
--- a/src/runtime/rpc/rpc_session.cc
+++ b/src/runtime/rpc/rpc_session.cc
@@ -149,7 +149,7 @@ class RPCSession::EventHandler {
TVMContext StripSessMask(TVMContext ctx) {
int dev_type = ctx.device_type;
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);
return ctx;
}
--
GitLab