From 0ee1d3f0bd65aeedd51b9ab86714ce25e1c6813e Mon Sep 17 00:00:00 2001 From: Chris Sullivan Date: Mon, 14 Jun 2021 21:32:54 -0700 Subject: [PATCH] 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 397f57b36dad..631d404845f7 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(workspace_->devices.size(), nullptr)});