mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-19 06:24:51 +08:00
Make tool for loading L0 builtins from binaries/spv
Change-Id: Ib7a50ecb85d740f13e6ed1b19e15bcd991b74867 Signed-off-by: Kacper Nowak <kacper.nowak@intel.com>
This commit is contained in:
11
shared/generate_cpp_array/CMakeLists.txt
Normal file
11
shared/generate_cpp_array/CMakeLists.txt
Normal file
@@ -0,0 +1,11 @@
|
||||
#
|
||||
# Copyright (C) 2020 Intel Corporation
|
||||
#
|
||||
# SPDX-License-Identifier: MIT
|
||||
#
|
||||
|
||||
set(CPP_GENERATE_TOOL_SOURCES
|
||||
${CMAKE_CURRENT_SOURCE_DIR}/CMakeLists.txt
|
||||
${CMAKE_CURRENT_SOURCE_DIR}/source/generate_cpp_array.cpp
|
||||
)
|
||||
add_executable (cpp_generate_tool "${CPP_GENERATE_TOOL_SOURCES}")
|
||||
115
shared/generate_cpp_array/source/generate_cpp_array.cpp
Normal file
115
shared/generate_cpp_array/source/generate_cpp_array.cpp
Normal file
@@ -0,0 +1,115 @@
|
||||
/*
|
||||
* Copyright (C) 2020 Intel Corporation
|
||||
*
|
||||
* SPDX-License-Identifier: MIT
|
||||
*
|
||||
*/
|
||||
|
||||
#include <fstream>
|
||||
#include <iomanip>
|
||||
#include <iostream>
|
||||
#include <sstream>
|
||||
|
||||
constexpr int ARG_COUNT = 9;
|
||||
|
||||
static void show_usage(std::string name) {
|
||||
std::cerr << "Usage " << name << "<option(s)> - ALL MUST BE SPECIFIED\n"
|
||||
<< "Options :\n"
|
||||
<< "\t -f, --file\t\tA file which content will be parsed into a uint32_t array in a .cpp file\n"
|
||||
<< "\t -o, --out\t\t.Cpp output file name\n"
|
||||
<< "\t -p, --platform\t\tFamily name with type\n"
|
||||
<< "\t -a, --array\t\tName of an uin32_t type array containing parsed input file" << std::endl;
|
||||
}
|
||||
std::string parseToCharArray(std::unique_ptr<uint8_t[]> binary, size_t size, std::string &builtinName, std::string &platform, bool isSpirV) {
|
||||
std::ostringstream out;
|
||||
|
||||
out << "#include <cstddef>\n";
|
||||
out << "#include <cstdint>\n\n";
|
||||
out << "size_t " << builtinName << "BinarySize_" << platform << " = " << size << ";\n";
|
||||
out << "uint32_t " << builtinName << "Binary_" << platform << "[" << (size + 3) / 4 << "] = {"
|
||||
<< std::endl
|
||||
<< " ";
|
||||
uint32_t *binaryUint = reinterpret_cast<uint32_t *>(binary.get());
|
||||
for (size_t i = 0; i < (size + 3) / 4; i++) {
|
||||
if (i != 0) {
|
||||
out << ", ";
|
||||
if (i % 8 == 0) {
|
||||
out << std::endl
|
||||
<< " ";
|
||||
}
|
||||
}
|
||||
if (i < size / 4) {
|
||||
out << "0x" << std::hex << std::setw(8) << std::setfill('0') << binaryUint[i];
|
||||
} else {
|
||||
uint32_t lastBytes = size & 0x3;
|
||||
uint32_t lastUint = 0;
|
||||
uint8_t *pLastUint = (uint8_t *)&lastUint;
|
||||
for (uint32_t j = 0; j < lastBytes; j++) {
|
||||
pLastUint[sizeof(uint32_t) - 1 - j] = binary.get()[i * 4 + j];
|
||||
}
|
||||
out << "0x" << std::hex << std::setw(8) << std::setfill('0') << lastUint;
|
||||
}
|
||||
}
|
||||
out << "};" << std::endl;
|
||||
|
||||
out << std::endl
|
||||
<< "#include \"shared/source/built_ins/registry/built_ins_registry.h\"\n"
|
||||
<< std::endl;
|
||||
out << "namespace NEO {" << std::endl;
|
||||
out << "static RegisterEmbeddedResource register" << builtinName;
|
||||
isSpirV ? out << "Ir(" : out << "Bin(";
|
||||
out << std::endl;
|
||||
out << " \"" << platform << "_0_" << builtinName;
|
||||
isSpirV ? out << ".builtin_kernel.spv\"," : out << ".builtin_kernel.bin\",";
|
||||
out << std::endl;
|
||||
out << " (const char *)" << builtinName << "Binary_" << platform << "," << std::endl;
|
||||
out << " " << builtinName << "BinarySize_" << platform << ");" << std::endl;
|
||||
out << "}" << std::endl;
|
||||
|
||||
return out.str();
|
||||
}
|
||||
|
||||
int main(int argc, char *argv[]) {
|
||||
if (argc != ARG_COUNT) {
|
||||
show_usage(argv[0]);
|
||||
return 1;
|
||||
}
|
||||
std::string fileName;
|
||||
std::string cppOutputName;
|
||||
std::string arrayName;
|
||||
std::string platform;
|
||||
size_t size = 0;
|
||||
std::fstream inputFile;
|
||||
bool isSpirV;
|
||||
for (int i = 1; i < argc; i++) {
|
||||
std::string arg = argv[i];
|
||||
if ((arg == "-f") || (arg == "--file")) {
|
||||
fileName = argv[++i];
|
||||
} else if ((arg == "-o") || (arg == "--output")) {
|
||||
cppOutputName = argv[++i];
|
||||
} else if ((arg == "-a") || (arg == "--array")) {
|
||||
arrayName = argv[++i];
|
||||
} else if ((arg == "-p") || (arg == "--platform")) {
|
||||
platform = argv[++i];
|
||||
} else {
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
inputFile.open(fileName.c_str(), std::ios::in | std::ios::binary | std::ios::ate);
|
||||
|
||||
if (inputFile.is_open()) {
|
||||
size = static_cast<size_t>(inputFile.tellg());
|
||||
std::unique_ptr<uint8_t[]> memblock = std::make_unique<uint8_t[]>(size);
|
||||
inputFile.clear();
|
||||
inputFile.seekg(0, std::ios::beg);
|
||||
inputFile.read(reinterpret_cast<char *>(memblock.get()), size);
|
||||
inputFile.close();
|
||||
isSpirV = fileName.find(".spv") != std::string::npos;
|
||||
std::string cpp = parseToCharArray(move(memblock), size, arrayName, platform, isSpirV);
|
||||
std::fstream(cppOutputName.c_str(), std::ios::out | std::ios::binary).write(cpp.c_str(), cpp.size());
|
||||
} else {
|
||||
std::cerr << "File cannot be opened!" << std::endl;
|
||||
return 1;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
Reference in New Issue
Block a user