[tvm] branch main updated: Update parsed kernel sources check. (#8257)

[masahi]

This is an automated email from the ASF dual-hosted git repository.

masahi pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new b85ac0e  Update parsed kernel sources check. (#8257)
b85ac0e is described below

commit b85ac0ef0f21de5528de695eec388eca98152347
Author: Chris Sullivan <csulli...@octoml.ai>
AuthorDate: Mon Jun 14 21:32:54 2021 -0700

    Update parsed kernel sources check. (#8257)
---
 src/runtime/opencl/opencl_module.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/runtime/opencl/opencl_module.cc 
b/src/runtime/opencl/opencl_module.cc
index 397f57b..631d404 100644
--- a/src/runtime/opencl/opencl_module.cc
+++ b/src/runtime/opencl/opencl_module.cc
@@ -193,8 +193,8 @@ void OpenCLModuleNode::Init() {
   ICHECK(!parsed_kernels_.empty()) << "The OpenCL module expects a kernel 
delimited "
                                    << "source from code generation, but no 
kernel "
                                    << "delimiter was found.";
-  ICHECK_EQ(workspace_->num_registered_kernels, parsed_kernels_.size())
-      << "The number of registered kernels does not match number of parsed 
kernel sources";
+  ICHECK_EQ(fmap_.size(), parsed_kernels_.size())
+      << "The number of parsed kernel sources does not match the number of 
kernel functions";
   // zero initialize cl_program pointers for each device kernel
   for (auto& kv : parsed_kernels_) {
     programs_.insert({kv.first, 
std::vector<cl_program>(workspace_->devices.size(), nullptr)});