Skip to content

Commit 5ee4091

Browse files
csullivantrevor-m
authored andcommittedJun 17, 2021
Update parsed kernel sources check. (apache#8257)
1 parent a316906 commit 5ee4091

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed
 

‎src/runtime/opencl/opencl_module.cc

+2-2
Original file line numberDiff line numberDiff line change
@@ -193,8 +193,8 @@ void OpenCLModuleNode::Init() {
193193
ICHECK(!parsed_kernels_.empty()) << "The OpenCL module expects a kernel delimited "
194194
<< "source from code generation, but no kernel "
195195
<< "delimiter was found.";
196-
ICHECK_EQ(workspace_->num_registered_kernels, parsed_kernels_.size())
197-
<< "The number of registered kernels does not match number of parsed kernel sources";
196+
ICHECK_EQ(fmap_.size(), parsed_kernels_.size())
197+
<< "The number of parsed kernel sources does not match the number of kernel functions";
198198
// zero initialize cl_program pointers for each device kernel
199199
for (auto& kv : parsed_kernels_) {
200200
programs_.insert({kv.first, std::vector<cl_program>(workspace_->devices.size(), nullptr)});

0 commit comments

Comments
 (0)