diff --git a/src/runtime/disco/disco_worker.cc b/src/runtime/disco/disco_worker.cc index e8ba351e791f..b281a3aca7da 100644 --- a/src/runtime/disco/disco_worker.cc +++ b/src/runtime/disco/disco_worker.cc @@ -77,7 +77,9 @@ struct DiscoWorker::Impl { } case DiscoAction::kCallPacked: { int func_reg_id = args[2]; + CHECK_LT(func_reg_id, self->register_file.size()); PackedFunc func = GetReg(self, func_reg_id); + CHECK(func.defined()); CallPacked(self, reg_id, func, TVMArgs(args.values + 3, args.type_codes + 3, args.num_args - 3)); break;