Method CreateFromProgram

tensorflow/lite/delegates/gpu/cl/cl_kernel.cc:106–126

}

Status CLKernel::CreateFromProgram(const CLProgram& program,
                                   const std::string& function_name) {
  int error_code;
  function_name_ = function_name;
  kernel_ =
      clCreateKernel(program.program(), function_name.c_str(), &error_code);
  if (!kernel_ || error_code != CL_SUCCESS) {
    kernel_ = nullptr;
    return UnknownError(absl::StrCat("Failed to create ", function_name,
                                     CLErrorCodeToString(error_code)));
  }

  program_ = program.program();
  clRetainProgram(program_);

  RETURN_IF_ERROR(GetKernelPrivateMemorySize(kernel_, program.GetDeviceId(),
                                             &private_memory_size_));
  RETURN_IF_ERROR(GetKernelMaxWorkGroupSize(kernel_, program.GetDeviceId(),
                                            &max_work_group_size_));
  return OkStatus();
}

Status CLKernel::SetMemory(int index, cl_mem memory) {
  return SetBytes(index, &memory, sizeof(cl_mem));