/*
|
* test-binary-kernel.cpp - Compile the source kernel into binary kernel
|
*
|
* Copyright (c) 2015 Intel Corporation
|
*
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
* you may not use this file except in compliance with the License.
|
* You may obtain a copy of the License at
|
*
|
* http://www.apache.org/licenses/LICENSE-2.0
|
*
|
* Unless required by applicable law or agreed to in writing, software
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
* See the License for the specific language governing permissions and
|
* limitations under the License.
|
*
|
* Author: Yinhang Liu <yinhangx.liu@intel.com>
|
*/
|
|
#include "test_common.h"
|
#include "test_inline.h"
|
#include "file_handle.h"
|
#include "ocl/cl_device.h"
|
#include "ocl/cl_context.h"
|
#include "ocl/cl_kernel.h"
|
#include <getopt.h>
|
|
using namespace XCam;
|
|
static void
|
print_help (const char *arg0)
|
{
|
printf ("Usage: %s --src-kernel <source-kernel> --bin-kernel <binary-kernel> --kernel-name <kernel-name>\n"
|
"\t --src-kernel specify source kernel path\n"
|
"\t --bin-kernel specify binary kernel path\n"
|
"\t --kernel-name specify kernel name\n"
|
"\t --help help\n"
|
, arg0);
|
}
|
|
#define FAILED_STATEMENT { \
|
if (kernel_body) xcam_free (kernel_body); \
|
if (kernel_name) xcam_free (kernel_name); \
|
if (program_binaries) xcam_free (program_binaries); \
|
return -1; }
|
|
int main (int argc, char *argv[])
|
{
|
char *src_path = NULL, *bin_path = NULL;
|
size_t src_size = 0;
|
size_t bin_size = 0;
|
char *kernel_name = NULL;
|
char *kernel_body = NULL;
|
uint8_t *program_binaries = NULL;
|
FileHandle src_file, bin_file;
|
XCamReturn ret = XCAM_RETURN_NO_ERROR;
|
|
const struct option long_opts [] = {
|
{"src-kernel", required_argument, NULL, 's'},
|
{"bin-kernel", required_argument, NULL, 'b'},
|
{"kernel-name", required_argument, NULL, 'n'},
|
{"help", no_argument, NULL, 'h'},
|
{NULL, 0, NULL, 0}
|
};
|
|
int opt = 0;
|
while ((opt = getopt_long (argc, argv, "", long_opts, NULL)) != -1) {
|
switch (opt) {
|
case 's':
|
src_path = optarg;
|
break;
|
case 'b':
|
bin_path = optarg;
|
break;
|
case 'n':
|
kernel_name = strndup (optarg, 1024);
|
break;
|
case 'h':
|
print_help (argv[0]);
|
return 0;
|
|
default:
|
print_help (argv[0]);
|
return -1;
|
}
|
}
|
|
if (!src_path || !bin_path) {
|
XCAM_LOG_ERROR ("path of source/binary kernel is null");
|
return -1;
|
}
|
if (!kernel_name) {
|
XCAM_LOG_ERROR ("kernel name is null");
|
return -1;
|
}
|
|
if (src_file.open (src_path, "r") != XCAM_RETURN_NO_ERROR ||
|
bin_file.open (bin_path, "wb") != XCAM_RETURN_NO_ERROR) {
|
XCAM_LOG_ERROR ("open source/binary kernel failed");
|
return -1;
|
}
|
|
ret = src_file.get_file_size (src_size);
|
CHECK_STATEMENT (ret, FAILED_STATEMENT, "get source sizes from %s failed", src_path);
|
|
kernel_body = (char *) xcam_malloc0 (sizeof (char) * (src_size + 1));
|
XCAM_ASSERT(kernel_body);
|
|
src_file.read_file (kernel_body, src_size);
|
CHECK_STATEMENT (ret, FAILED_STATEMENT, "read source from %s failed", src_path);
|
kernel_body[src_size] = '\0';
|
|
SmartPtr<CLContext> context;
|
context = CLDevice::instance ()->get_context ();
|
SmartPtr<CLKernel> kernel = new CLKernel (context, kernel_name);
|
kernel->load_from_source (kernel_body, strlen (kernel_body), &program_binaries, &bin_size);
|
|
ret = bin_file.write_file (program_binaries, bin_size);
|
CHECK_STATEMENT (ret, FAILED_STATEMENT, "write binary to %s failed", bin_path);
|
|
xcam_free (kernel_name);
|
xcam_free (kernel_body);
|
xcam_free (program_binaries);
|
return 0;
|
}
|
