/* * Copyright (C) 2018-2023 Intel Corporation * * SPDX-License-Identifier: MIT * */ #include "offline_compiler.h" #include "shared/offline_compiler/source/ocloc_error_code.h" #include "shared/offline_compiler/source/queries.h" #include "shared/offline_compiler/source/utilities/get_git_version_info.h" #include "shared/source/compiler_interface/compiler_options.h" #include "shared/source/compiler_interface/default_cache_config.h" #include "shared/source/compiler_interface/intermediate_representations.h" #include "shared/source/debug_settings/debug_settings_manager.h" #include "shared/source/device_binary_format/device_binary_formats.h" #include "shared/source/device_binary_format/elf/elf_encoder.h" #include "shared/source/device_binary_format/elf/ocl_elf.h" #include "shared/source/helpers/compiler_options_parser.h" #include "shared/source/helpers/compiler_product_helper.h" #include "shared/source/helpers/debug_helpers.h" #include "shared/source/helpers/string.h" #include "shared/source/helpers/validators.h" #include "shared/source/os_interface/os_library.h" #include "platforms.h" #include #include #include #ifdef _WIN32 #include #define MakeDirectory _mkdir #define GetCurrentWorkingDirectory _getcwd #else #include #define MakeDirectory(dir) mkdir(dir, 0777) #define GetCurrentWorkingDirectory getcwd #endif using namespace NEO::OclocErrorCode; namespace NEO { std::string convertToPascalCase(const std::string &inString) { std::string outString; bool capitalize = true; for (unsigned int i = 0; i < inString.length(); i++) { if (isalpha(inString[i]) && capitalize == true) { outString += toupper(inString[i]); capitalize = false; } else if (inString[i] == '_') { capitalize = true; } else { outString += inString[i]; } } return outString; } std::string getDeprecatedDevices(OclocArgHelper *helper) { auto acronyms = helper->productConfigHelper->getDeprecatedAcronyms(); return helper->createStringForArgs(acronyms); } std::string getSupportedDevices(OclocArgHelper *helper) { auto devices = helper->productConfigHelper->getAllProductAcronyms(); auto families = helper->productConfigHelper->getFamiliesAcronyms(); auto releases = helper->productConfigHelper->getReleasesAcronyms(); auto familiesAndReleases = helper->getArgsWithoutDuplicate(families, releases); return helper->createStringForArgs(devices, familiesAndReleases); } OfflineCompiler::OfflineCompiler() = default; OfflineCompiler::~OfflineCompiler() { pBuildInfo.reset(); delete[] irBinary; delete[] genBinary; delete[] debugDataBinary; } OfflineCompiler *OfflineCompiler::create(size_t numArgs, const std::vector &allArgs, bool dumpFiles, int &retVal, OclocArgHelper *helper) { retVal = SUCCESS; auto pOffCompiler = new OfflineCompiler(); if (pOffCompiler) { pOffCompiler->argHelper = helper; pOffCompiler->fclFacade = std::make_unique(helper); pOffCompiler->igcFacade = std::make_unique(helper); retVal = pOffCompiler->initialize(numArgs, allArgs, dumpFiles); } if (retVal != SUCCESS) { delete pOffCompiler; pOffCompiler = nullptr; } return pOffCompiler; } void printQueryHelp(OclocArgHelper *helper) { helper->printf(OfflineCompiler::queryHelp.data()); } int OfflineCompiler::query(size_t numArgs, const std::vector &allArgs, OclocArgHelper *helper) { if (allArgs.size() != 3) { helper->printf("Error: Invalid command line. Expected ocloc query "); return INVALID_COMMAND_LINE; } auto retVal = SUCCESS; auto &arg = allArgs[2]; if (Queries::queryNeoRevision == arg) { auto revision = NEO::getRevision(); helper->saveOutput(Queries::queryNeoRevision.data(), revision.c_str(), revision.size() + 1); } else if (Queries::queryOCLDriverVersion == arg) { auto driverVersion = NEO::getOclDriverVersion(); helper->saveOutput(Queries::queryOCLDriverVersion.data(), driverVersion.c_str(), driverVersion.size() + 1); } else if ("--help" == arg) { printQueryHelp(helper); } else { helper->printf("Error: Invalid command line. Unknown argument %s.", arg.c_str()); retVal = INVALID_COMMAND_LINE; } return retVal; } void printAcronymIdsHelp(OclocArgHelper *helper) { helper->printf(OfflineCompiler::idsHelp.data(), getSupportedDevices(helper).c_str()); } int OfflineCompiler::queryAcronymIds(size_t numArgs, const std::vector &allArgs, OclocArgHelper *helper) { const size_t numArgRequested = 3u; auto retVal = SUCCESS; if (numArgs != numArgRequested) { helper->printf("Error: Invalid command line. Expected ocloc ids .\n"); retVal = INVALID_COMMAND_LINE; return retVal; } else if ((ConstStringRef("-h") == allArgs[2] || ConstStringRef("--help") == allArgs[2])) { printAcronymIdsHelp(helper); return retVal; } std::string queryAcronym = allArgs[2]; ProductConfigHelper::adjustDeviceName(queryAcronym); auto &enabledDevices = helper->productConfigHelper->getDeviceAotInfo(); std::vector matchedVersions{}; if (helper->productConfigHelper->isFamily(queryAcronym)) { auto family = ProductConfigHelper::getFamilyForAcronym(queryAcronym); for (const auto &device : enabledDevices) { if (device.family == family) { matchedVersions.push_back(ProductConfigHelper::parseMajorMinorRevisionValue(device.aotConfig)); } } } else if (helper->productConfigHelper->isRelease(queryAcronym)) { auto release = ProductConfigHelper::getReleaseForAcronym(queryAcronym); for (const auto &device : enabledDevices) { if (device.release == release) { matchedVersions.push_back(ProductConfigHelper::parseMajorMinorRevisionValue(device.aotConfig)); } } } else if (helper->productConfigHelper->isProductConfig(queryAcronym)) { auto product = ProductConfigHelper::getProductConfigForAcronym(queryAcronym); for (const auto &device : enabledDevices) { if (device.aotConfig.value == product) { matchedVersions.push_back(ProductConfigHelper::parseMajorMinorRevisionValue(device.aotConfig)); } } } else { helper->printf("Error: Invalid command line. Unknown acronym %s.\n", allArgs[2].c_str()); retVal = INVALID_COMMAND_LINE; return retVal; } std::ostringstream os; for (const auto &prefix : matchedVersions) { if (os.tellp()) os << "\n"; os << prefix; } helper->printf("Matched ids:\n%s\n", os.str().c_str()); return retVal; } struct OfflineCompiler::buildInfo { std::unique_ptr> fclOptions; std::unique_ptr> fclInternalOptions; std::unique_ptr> fclOutput; IGC::CodeType::CodeType_t intermediateRepresentation; }; int OfflineCompiler::buildIrBinary() { int retVal = SUCCESS; if (allowCaching) { irHash = cache->getCachedFileName(getHardwareInfo(), sourceCode, options, internalOptions); irBinary = cache->loadCachedBinary(irHash, irBinarySize).release(); if (irBinary) { return retVal; } } UNRECOVERABLE_IF(!fclFacade->isInitialized()); pBuildInfo->intermediateRepresentation = useLlvmText ? IGC::CodeType::llvmLl : (useLlvmBc ? IGC::CodeType::llvmBc : preferredIntermediateRepresentation); // sourceCode.size() returns the number of characters without null terminated char CIF::RAII::UPtr_t fclSrc = nullptr; pBuildInfo->fclOptions = fclFacade->createConstBuffer(options.c_str(), options.size()); pBuildInfo->fclInternalOptions = fclFacade->createConstBuffer(internalOptions.c_str(), internalOptions.size()); auto err = fclFacade->createConstBuffer(nullptr, 0); auto srcType = IGC::CodeType::undefined; std::vector tempSrcStorage; if (this->argHelper->hasHeaders()) { srcType = IGC::CodeType::elf; NEO::Elf::ElfEncoder<> elfEncoder(true, true, 1U); elfEncoder.getElfFileHeader().type = NEO::Elf::ET_OPENCL_SOURCE; elfEncoder.appendSection(NEO::Elf::SHT_OPENCL_SOURCE, "CLMain", sourceCode); for (const auto &header : this->argHelper->getHeaders()) { ArrayRef headerData(header.data, header.length); ConstStringRef headerName = header.name; elfEncoder.appendSection(NEO::Elf::SHT_OPENCL_HEADER, headerName, headerData); } tempSrcStorage = elfEncoder.encode(); fclSrc = fclFacade->createConstBuffer(tempSrcStorage.data(), tempSrcStorage.size()); } else { srcType = IGC::CodeType::oclC; fclSrc = fclFacade->createConstBuffer(sourceCode.c_str(), sourceCode.size() + 1); } auto fclTranslationCtx = fclFacade->createTranslationContext(srcType, pBuildInfo->intermediateRepresentation, err.get()); if (true == NEO::areNotNullptr(err->GetMemory())) { updateBuildLog(err->GetMemory(), err->GetSizeRaw()); retVal = BUILD_PROGRAM_FAILURE; return retVal; } if (false == NEO::areNotNullptr(fclSrc.get(), pBuildInfo->fclOptions.get(), pBuildInfo->fclInternalOptions.get(), fclTranslationCtx.get())) { retVal = OUT_OF_HOST_MEMORY; return retVal; } pBuildInfo->fclOutput = fclTranslationCtx->Translate(fclSrc.get(), pBuildInfo->fclOptions.get(), pBuildInfo->fclInternalOptions.get(), nullptr, 0); if (pBuildInfo->fclOutput == nullptr) { retVal = OUT_OF_HOST_MEMORY; return retVal; } UNRECOVERABLE_IF(pBuildInfo->fclOutput->GetBuildLog() == nullptr); UNRECOVERABLE_IF(pBuildInfo->fclOutput->GetOutput() == nullptr); if (pBuildInfo->fclOutput->Successful() == false) { updateBuildLog(pBuildInfo->fclOutput->GetBuildLog()->GetMemory(), pBuildInfo->fclOutput->GetBuildLog()->GetSizeRaw()); retVal = BUILD_PROGRAM_FAILURE; return retVal; } storeBinary(irBinary, irBinarySize, pBuildInfo->fclOutput->GetOutput()->GetMemory(), pBuildInfo->fclOutput->GetOutput()->GetSizeRaw()); isSpirV = pBuildInfo->intermediateRepresentation == IGC::CodeType::spirV; updateBuildLog(pBuildInfo->fclOutput->GetBuildLog()->GetMemory(), pBuildInfo->fclOutput->GetBuildLog()->GetSizeRaw()); if (allowCaching) { cache->cacheBinary(irHash, irBinary, static_cast(irBinarySize)); } return retVal; } std::string OfflineCompiler::validateInputType(const std::string &input, bool isLlvm, bool isSpirv) { auto asBitcode = ArrayRef::fromAny(input.data(), input.size()); if (isSpirv) { if (NEO::isSpirVBitcode(asBitcode)) { return ""; } return "Warning : file does not look like spirv bitcode (wrong magic numbers)"; } if (isLlvm) { if (NEO::isLlvmBitcode(asBitcode)) { return ""; } return "Warning : file does not look like llvm bitcode (wrong magic numbers)"; } if (NEO::isSpirVBitcode(asBitcode)) { return "Warning : file looks like spirv bitcode (based on magic numbers) - please make sure proper CLI flags are present"; } if (NEO::isLlvmBitcode(asBitcode)) { return "Warning : file looks like llvm bitcode (based on magic numbers) - please make sure proper CLI flags are present"; } return ""; } int OfflineCompiler::buildSourceCode() { int retVal = SUCCESS; if (sourceCode.empty()) { return INVALID_PROGRAM; } if (allowCaching) { irHash = cache->getCachedFileName(getHardwareInfo(), sourceCode, options, internalOptions); irBinary = cache->loadCachedBinary(irHash, irBinarySize).release(); genHash = cache->getCachedFileName(getHardwareInfo(), ArrayRef(irBinary, irBinarySize), options, internalOptions); genBinary = cache->loadCachedBinary(genHash, genBinarySize).release(); if (irBinary && genBinary) { if (!CompilerOptions::contains(options, CompilerOptions::generateDebugInfo)) return retVal; else { dbgHash = cache->getCachedFileName(getHardwareInfo(), irHash, options, internalOptions); debugDataBinary = cache->loadCachedBinary(dbgHash, debugDataBinarySize).release(); if (debugDataBinary) return retVal; } } } UNRECOVERABLE_IF(!igcFacade->isInitialized()); auto inputTypeWarnings = validateInputType(sourceCode, inputFileLlvm, inputFileSpirV); this->argHelper->printf(inputTypeWarnings.c_str()); CIF::RAII::UPtr_t igcOutput; bool inputIsIntermediateRepresentation = inputFileLlvm || inputFileSpirV; if (false == inputIsIntermediateRepresentation) { retVal = buildIrBinary(); if (retVal != SUCCESS) return retVal; auto igcTranslationCtx = igcFacade->createTranslationContext(pBuildInfo->intermediateRepresentation, IGC::CodeType::oclGenBin); igcOutput = igcTranslationCtx->Translate(pBuildInfo->fclOutput->GetOutput(), pBuildInfo->fclOptions.get(), pBuildInfo->fclInternalOptions.get(), nullptr, 0); } else { storeBinary(irBinary, irBinarySize, sourceCode.c_str(), sourceCode.size()); isSpirV = inputFileSpirV; auto igcSrc = igcFacade->createConstBuffer(sourceCode.c_str(), sourceCode.size()); auto igcOptions = igcFacade->createConstBuffer(options.c_str(), options.size()); auto igcInternalOptions = igcFacade->createConstBuffer(internalOptions.c_str(), internalOptions.size()); auto igcTranslationCtx = igcFacade->createTranslationContext(inputFileSpirV ? IGC::CodeType::spirV : IGC::CodeType::llvmBc, IGC::CodeType::oclGenBin); igcOutput = igcTranslationCtx->Translate(igcSrc.get(), igcOptions.get(), igcInternalOptions.get(), nullptr, 0); } if (igcOutput == nullptr) { return OUT_OF_HOST_MEMORY; } UNRECOVERABLE_IF(igcOutput->GetBuildLog() == nullptr); UNRECOVERABLE_IF(igcOutput->GetOutput() == nullptr); updateBuildLog(igcOutput->GetBuildLog()->GetMemory(), igcOutput->GetBuildLog()->GetSizeRaw()); if (igcOutput->GetOutput()->GetSizeRaw() != 0) { storeBinary(genBinary, genBinarySize, igcOutput->GetOutput()->GetMemory(), igcOutput->GetOutput()->GetSizeRaw()); } if (igcOutput->GetDebugData()->GetSizeRaw() != 0) { storeBinary(debugDataBinary, debugDataBinarySize, igcOutput->GetDebugData()->GetMemory(), igcOutput->GetDebugData()->GetSizeRaw()); } if (allowCaching) { cache->cacheBinary(irHash, irBinary, static_cast(irBinarySize)); cache->cacheBinary(genHash, genBinary, static_cast(genBinarySize)); cache->cacheBinary(dbgHash, debugDataBinary, static_cast(debugDataBinarySize)); } retVal = igcOutput->Successful() ? SUCCESS : BUILD_PROGRAM_FAILURE; return retVal; } int OfflineCompiler::build() { int retVal = SUCCESS; if (isOnlySpirV()) { retVal = buildIrBinary(); } else { retVal = buildSourceCode(); } generateElfBinary(); if (dumpFiles) { writeOutAllFiles(); } return retVal; } void OfflineCompiler::updateBuildLog(const char *pErrorString, const size_t errorStringSize) { if (pErrorString != nullptr) { std::string log(pErrorString, pErrorString + errorStringSize); ConstStringRef errorString(log); const bool warningFound = errorString.containsCaseInsensitive("warning"); if (!isQuiet() || !warningFound) { if (buildLog.empty()) { buildLog.assign(errorString.data()); } else { buildLog.append("\n"); buildLog.append(errorString.data()); } } } } std::string &OfflineCompiler::getBuildLog() { return buildLog; } int OfflineCompiler::initHardwareInfoForDeprecatedAcronyms(std::string deviceName, std::unique_ptr &&compilerProductHelper) { std::vector allSupportedProduct{ALL_SUPPORTED_PRODUCT_FAMILIES}; std::transform(deviceName.begin(), deviceName.end(), deviceName.begin(), ::tolower); for (const auto &product : allSupportedProduct) { if (0 == strcmp(deviceName.c_str(), hardwarePrefix[product])) { hwInfo = *hardwareInfoTable[product]; if (revisionId != -1) { hwInfo.platform.usRevId = revisionId; } compilerProductHelper = NEO::CompilerProductHelper::create(hwInfo.platform.eProductFamily); uint64_t config = hwInfoConfig ? hwInfoConfig : compilerProductHelper->getHwInfoConfig(hwInfo); setHwInfoValuesFromConfig(config, hwInfo); hardwareInfoBaseSetup[hwInfo.platform.eProductFamily](&hwInfo, true); UNRECOVERABLE_IF(compilerProductHelper == nullptr); productFamilyName = hardwarePrefix[hwInfo.platform.eProductFamily]; return SUCCESS; } } return INVALID_DEVICE; } bool OfflineCompiler::isArgumentDeviceId(const std::string &argument) const { const char hexPrefixLength = 2; return argument.substr(0, hexPrefixLength) == "0x" && std::all_of(argument.begin() + hexPrefixLength, argument.end(), (::isxdigit)); } int OfflineCompiler::initHardwareInfoForProductConfig(std::string deviceName) { AOT::PRODUCT_CONFIG productConfig = AOT::UNKNOWN_ISA; ProductConfigHelper::adjustDeviceName(deviceName); uint32_t deviceID = 0; if (deviceName.find(".") != std::string::npos) { productConfig = argHelper->productConfigHelper->getProductConfigForVersionValue(deviceName); } else if (isArgumentDeviceId(deviceName)) { deviceID = std::stoi(deviceName, 0, 16); productConfig = argHelper->productConfigHelper->getProductConfigForDeviceId(deviceID); } else if (argHelper->productConfigHelper->isProductConfig(deviceName)) { productConfig = ProductConfigHelper::getProductConfigForAcronym(deviceName); } else { return INVALID_DEVICE; } if (argHelper->getHwInfoForProductConfig(productConfig, hwInfo, hwInfoConfig, deviceID, revisionId, std::move(compilerProductHelper))) { if (deviceID) { auto product = argHelper->productConfigHelper->getAcronymForProductConfig(productConfig); argHelper->printf("Auto-detected target based on %s device id: %s\n", deviceName.c_str(), product.c_str()); } deviceConfig = productConfig; productFamilyName = hardwarePrefix[hwInfo.platform.eProductFamily]; return SUCCESS; } else { argHelper->printf("Could not determine device target: %s\n", deviceName.c_str()); return INVALID_DEVICE; } } int OfflineCompiler::initHardwareInfo(std::string deviceName) { int retVal = INVALID_DEVICE; if (deviceName.empty()) { return retVal; } retVal = initHardwareInfoForProductConfig(deviceName); if (retVal == SUCCESS) { return retVal; } retVal = initHardwareInfoForDeprecatedAcronyms(deviceName, std::move(compilerProductHelper)); return retVal; } std::string OfflineCompiler::getStringWithinDelimiters(const std::string &src) { size_t start = src.find("R\"===("); size_t stop = src.find(")===\""); DEBUG_BREAK_IF(std::string::npos == start); DEBUG_BREAK_IF(std::string::npos == stop); start += strlen("R\"===("); size_t size = stop - start; std::string dst(src, start, size + 1); dst[size] = '\0'; // put null char at the end return dst; } int OfflineCompiler::initialize(size_t numArgs, const std::vector &allArgs, bool dumpFiles) { this->dumpFiles = dumpFiles; int retVal = SUCCESS; const char *source = nullptr; std::unique_ptr sourceFromFile; size_t sourceFromFileSize = 0; this->pBuildInfo = std::make_unique(); auto cacheConfig = getDefaultCompilerCacheConfig(); cacheDir = cacheConfig.cacheDir; retVal = parseCommandLine(numArgs, allArgs); if (showHelp) { printUsage(); return retVal; } else if (retVal != SUCCESS) { return retVal; } if (options.empty()) { // try to read options from file if not provided by commandline size_t extStart = inputFile.find_last_of("."); if (extStart != std::string::npos) { std::string oclocOptionsFileName = inputFile.substr(0, extStart); oclocOptionsFileName.append("_ocloc_options.txt"); std::string oclocOptionsFromFile; bool oclocOptionsRead = readOptionsFromFile(oclocOptionsFromFile, oclocOptionsFileName, argHelper); if (oclocOptionsRead) { if (!isQuiet()) { argHelper->printf("Building with ocloc options:\n%s\n", oclocOptionsFromFile.c_str()); } std::istringstream iss(allArgs[0] + " " + oclocOptionsFromFile); std::vector tokens{ std::istream_iterator{iss}, std::istream_iterator{}}; retVal = parseCommandLine(tokens.size(), tokens); if (retVal != SUCCESS) { argHelper->printf("Failed with ocloc options from file:\n%s\n", oclocOptionsFromFile.c_str()); return retVal; } } std::string optionsFileName = inputFile.substr(0, extStart); optionsFileName.append("_options.txt"); bool optionsRead = readOptionsFromFile(options, optionsFileName, argHelper); if (optionsRead) { optionsReadFromFile = std::string(options); } if (optionsRead && !isQuiet()) { argHelper->printf("Building with options:\n%s\n", options.c_str()); } std::string internalOptionsFileName = inputFile.substr(0, extStart); internalOptionsFileName.append("_internal_options.txt"); std::string internalOptionsFromFile; bool internalOptionsRead = readOptionsFromFile(internalOptionsFromFile, internalOptionsFileName, argHelper); if (internalOptionsRead) { internalOptionsReadFromFile = std::string(internalOptionsFromFile); } if (internalOptionsRead && !isQuiet()) { argHelper->printf("Building with internal options:\n%s\n", internalOptionsFromFile.c_str()); } CompilerOptions::concatenateAppend(internalOptions, internalOptionsFromFile); } } retVal = deviceName.empty() ? SUCCESS : initHardwareInfo(deviceName.c_str()); if (retVal != SUCCESS) { argHelper->printf("Error: Cannot get HW Info for device %s.\n", deviceName.c_str()); return retVal; } if (!formatToEnforce.empty()) { enforceFormat(formatToEnforce); } if (CompilerOptions::contains(options, CompilerOptions::generateDebugInfo.str())) { if (false == inputFileSpirV && false == CompilerOptions::contains(options, CompilerOptions::generateSourcePath) && false == CompilerOptions::contains(options, CompilerOptions::useCMCompiler)) { auto sourcePathStringOption = CompilerOptions::generateSourcePath.str(); sourcePathStringOption.append(" "); sourcePathStringOption.append(CompilerOptions::wrapInQuotes(inputFile)); options = CompilerOptions::concatenate(options, sourcePathStringOption); } } if (deviceName.empty()) { internalOptions = CompilerOptions::concatenate("-ocl-version=300 -cl-ext=-all,+cl_khr_3d_image_writes", internalOptions); CompilerOptions::concatenateAppend(internalOptions, CompilerOptions::enableImageSupport); } else { appendExtensionsToInternalOptions(hwInfo, options, internalOptions); appendExtraInternalOptions(internalOptions); } parseDebugSettings(); // set up the device inside the program sourceFromFile = argHelper->loadDataFromFile(inputFile, sourceFromFileSize); if (sourceFromFileSize == 0) { retVal = INVALID_FILE; return retVal; } if (inputFileLlvm || inputFileSpirV) { // use the binary input "as is" sourceCode.assign(sourceFromFile.get(), sourceFromFileSize); } else { // for text input, we also accept files used as runtime builtins source = strstr((const char *)sourceFromFile.get(), "R\"===("); sourceCode = (source != nullptr) ? getStringWithinDelimiters(sourceFromFile.get()) : sourceFromFile.get(); } if ((inputFileSpirV == false) && (inputFileLlvm == false)) { const auto fclInitializationResult = fclFacade->initialize(hwInfo); if (fclInitializationResult != SUCCESS) { argHelper->printf("Error! FCL initialization failure. Error code = %d\n", fclInitializationResult); return fclInitializationResult; } preferredIntermediateRepresentation = fclFacade->getPreferredIntermediateRepresentation(); } else { if (!isQuiet()) { argHelper->printf("Compilation from IR - skipping loading of FCL\n"); } preferredIntermediateRepresentation = IGC::CodeType::spirV; } const auto igcInitializationResult = igcFacade->initialize(hwInfo); if (igcInitializationResult != SUCCESS) { argHelper->printf("Error! IGC initialization failure. Error code = %d\n", igcInitializationResult); return igcInitializationResult; } if (allowCaching) { cacheConfig.cacheDir = cacheDir; cache = std::make_unique(cacheConfig); createDir(cacheConfig.cacheDir); } return retVal; } int OfflineCompiler::parseCommandLine(size_t numArgs, const std::vector &argv) { int retVal = SUCCESS; bool compile32 = false; bool compile64 = false; for (uint32_t argIndex = 1; argIndex < argv.size(); argIndex++) { const auto &currArg = argv[argIndex]; const bool hasMoreArgs = (argIndex + 1 < numArgs); if ("compile" == currArg) { // skip it } else if (("-file" == currArg) && hasMoreArgs) { inputFile = argv[argIndex + 1]; argIndex++; } else if (("-output" == currArg) && hasMoreArgs) { outputFile = argv[argIndex + 1]; argIndex++; } else if ((CompilerOptions::arch32bit == currArg) || ("-32" == currArg)) { compile32 = true; CompilerOptions::concatenateAppend(internalOptions, CompilerOptions::arch32bit); } else if ((CompilerOptions::arch64bit == currArg) || ("-64" == currArg)) { compile64 = true; CompilerOptions::concatenateAppend(internalOptions, CompilerOptions::arch64bit); } else if (CompilerOptions::greaterThan4gbBuffersRequired == currArg) { CompilerOptions::concatenateAppend(internalOptions, CompilerOptions::greaterThan4gbBuffersRequired); } else if (("-device" == currArg) && hasMoreArgs) { deviceName = argv[argIndex + 1]; argIndex++; } else if ("-llvm_text" == currArg) { useLlvmText = true; } else if ("-llvm_bc" == currArg) { useLlvmBc = true; } else if ("-llvm_input" == currArg) { inputFileLlvm = true; } else if ("-spirv_input" == currArg) { inputFileSpirV = true; } else if ("-cpp_file" == currArg) { useCppFile = true; } else if ("-gen_file" == currArg) { useGenFile = true; } else if (("-options" == currArg) && hasMoreArgs) { CompilerOptions::concatenateAppend(options, argv[argIndex + 1]); argIndex++; } else if (("-internal_options" == currArg) && hasMoreArgs) { CompilerOptions::concatenateAppend(internalOptions, argv[argIndex + 1]); argIndex++; } else if ("-options_name" == currArg) { useOptionsSuffix = true; } else if ("-force_stos_opt" == currArg) { forceStatelessToStatefulOptimization = true; } else if (("-out_dir" == currArg) && hasMoreArgs) { outputDirectory = argv[argIndex + 1]; argIndex++; } else if (("-cache_dir" == currArg) && hasMoreArgs) { cacheDir = argv[argIndex + 1]; argIndex++; } else if ("-q" == currArg) { quiet = true; } else if ("-qq" == currArg) { argHelper->getPrinterRef() = MessagePrinter(true); quiet = true; } else if ("-spv_only" == currArg) { onlySpirV = true; } else if ("-output_no_suffix" == currArg) { outputNoSuffix = true; } else if ("--help" == currArg) { showHelp = true; return SUCCESS; } else if (("-revision_id" == currArg) && hasMoreArgs) { revisionId = std::stoi(argv[argIndex + 1], nullptr, 0); argIndex++; } else if ("-exclude_ir" == currArg) { excludeIr = true; } else if ("--format" == currArg) { formatToEnforce = argv[argIndex + 1]; argIndex++; } else if (("-config" == currArg) && hasMoreArgs) { parseHwInfoConfigString(argv[argIndex + 1], hwInfoConfig); if (!hwInfoConfig) { argHelper->printf("Error: Invalid config.\n"); retVal = INVALID_COMMAND_LINE; break; } argIndex++; } else if ("-allow_caching" == currArg) { allowCaching = true; } else { argHelper->printf("Invalid option (arg %d): %s\n", argIndex, argv[argIndex].c_str()); retVal = INVALID_COMMAND_LINE; break; } } unifyExcludeIrFlags(); if (DebugManager.flags.OverrideRevision.get() != -1) { revisionId = static_cast(DebugManager.flags.OverrideRevision.get()); } if (retVal == SUCCESS) { if (compile32 && compile64) { argHelper->printf("Error: Cannot compile for 32-bit and 64-bit, please choose one.\n"); retVal |= INVALID_COMMAND_LINE; } if (deviceName.empty() && (false == onlySpirV)) { argHelper->printf("Error: Device name missing.\n"); retVal = INVALID_COMMAND_LINE; } if (inputFile.empty()) { argHelper->printf("Error: Input file name missing.\n"); retVal = INVALID_COMMAND_LINE; } else if (!argHelper->fileExists(inputFile)) { argHelper->printf("Error: Input file %s missing.\n", inputFile.c_str()); retVal = INVALID_FILE; } } return retVal; } void OfflineCompiler::unifyExcludeIrFlags() { const auto excludeIrFromZebin{internalOptions.find(CompilerOptions::excludeIrFromZebin.data()) != std::string::npos}; if (!excludeIr && excludeIrFromZebin) { excludeIr = true; } else if (excludeIr && !excludeIrFromZebin) { const std::string prefix{"-ze"}; CompilerOptions::concatenateAppend(internalOptions, prefix + CompilerOptions::excludeIrFromZebin.data()); } } void OfflineCompiler::setStatelessToStatefulBufferOffsetFlag() { bool isStatelessToStatefulBufferOffsetSupported = true; if (!deviceName.empty()) { isStatelessToStatefulBufferOffsetSupported = compilerProductHelper->isStatelessToStatefulBufferOffsetSupported(); } if (DebugManager.flags.EnableStatelessToStatefulBufferOffsetOpt.get() != -1) { isStatelessToStatefulBufferOffsetSupported = DebugManager.flags.EnableStatelessToStatefulBufferOffsetOpt.get() != 0; } if (isStatelessToStatefulBufferOffsetSupported) { CompilerOptions::concatenateAppend(internalOptions, CompilerOptions::hasBufferOffsetArg); } } void OfflineCompiler::appendExtraInternalOptions(std::string &internalOptions) { if (compilerProductHelper->isForceToStatelessRequired() && !forceStatelessToStatefulOptimization) { CompilerOptions::concatenateAppend(internalOptions, CompilerOptions::greaterThan4gbBuffersRequired); } if (compilerProductHelper->isForceEmuInt32DivRemSPRequired()) { CompilerOptions::concatenateAppend(internalOptions, CompilerOptions::forceEmuInt32DivRemSP); } CompilerOptions::concatenateAppend(internalOptions, compilerProductHelper->getCachingPolicyOptions(false)); } void OfflineCompiler::parseDebugSettings() { setStatelessToStatefulBufferOffsetFlag(); } std::string OfflineCompiler::parseBinAsCharArray(uint8_t *binary, size_t size, std::string &fileName) { std::string builtinName = convertToPascalCase(fileName); std::ostringstream out; // Convert binary to cpp out << "#include \n"; out << "#include \n\n"; out << "size_t " << builtinName << "BinarySize_" << productFamilyName << " = " << size << ";\n"; out << "uint32_t " << builtinName << "Binary_" << productFamilyName << "[" << (size + 3) / 4 << "] = {" << std::endl << " "; uint32_t *binaryUint = (uint32_t *)binary; 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[i * 4 + j]; } out << "0x" << std::hex << std::setw(8) << std::setfill('0') << lastUint; } } out << "};" << std::endl; return out.str(); } std::string OfflineCompiler::getFileNameTrunk(std::string &filePath) { size_t slashPos = filePath.find_last_of("\\/", filePath.size()) + 1; size_t extPos = filePath.find_last_of(".", filePath.size()); if (extPos == std::string::npos) { extPos = filePath.size(); } std::string fileTrunk = filePath.substr(slashPos, (extPos - slashPos)); return fileTrunk; } void OfflineCompiler::printUsage() { argHelper->printf(R"===(Compiles input file to Intel Compute GPU device binary (*.bin). Additionally, outputs intermediate representation (e.g. spirV). Different input and intermediate file formats are available. Usage: ocloc [compile] -file -device [-output ] [-out_dir ] [-options ] [-32|-64] [-internal_options ] [-llvm_text|-llvm_input|-spirv_input] [-options_name] [-q] [-cpp_file] [-output_no_suffix] [--help] -file The input file to be compiled (by default input source format is OpenCL C kernel language). -device Target device. can be: %s, version or hexadecimal value with 0x prefix - can be single or multiple target devices. The version is a representation of the .. value. The hexadecimal value represents device ID. If such value is provided, ocloc will try to match it with corresponding device type. For example, 0xFF20 device ID will be translated to tgllp. If multiple target devices are provided, ocloc will compile for each of these targets and will create a fatbinary archive that contains all of device binaries produced this way. Supported -device patterns examples: -device 0x4905 ; will compile 1 target (dg1) -device 12.10.0 ; will compile 1 target (dg1) -device dg1 ; will compile 1 target -device dg1,acm-g10 ; will compile 2 targets -device dg1:acm-g10 ; will compile all targets in range (inclusive) -device dg1: ; will compile all targets newer/same as provided -device :dg1 ; will compile all targets older/same as provided -device xe-hpg ; will compile all targets matching the same release -device xe ; will compile all targets matching the same family -device xe-hpg:xe-hpc ; will compile all targets in range (inclusive) -device xe-hpg: ; will compile all targets newer/same as provided -device :xe-hpg ; will compile all targets older/same as provided known to ocloc Deprecated notation that is still supported: can be: %s - can be single target device. -output Optional output file base name. Default is input file's base name. This base name will be used for all output files. Proper sufixes (describing file formats) will be added automatically. -out_dir Optional output directory. Default is current working directory. -allow_caching Allows caching binaries from compilation (like spirv, gen or debug data) and loading them by ocloc when the same program is compiled again. -cache_dir Optional caching directory. Default directory is "ocloc_cache". -options Optional OpenCL C compilation options as defined by OpenCL specification. Special options for Vector Compute: -vc-codegen compile from SPIRV -cmc compile from CM sources -32 Forces target architecture to 32-bit pointers. Default pointer size is inherited from ocloc's pointer size. This option is exclusive with -64. -64 Forces target architecture to 64-bit pointers. Default pointer size is inherited from ocloc's pointer size. This option is exclusive with -32. -internal_options Optional compiler internal options as defined by compilers used underneath. Check intel-graphics-compiler (IGC) project for details on available internal options. You also may provide explicit --help to inquire information about option, mentioned in -options -llvm_text Forces intermediate representation (IR) format to human-readable LLVM IR (.ll). This option affects only output files and should not be used in combination with '-llvm_input' option. Default IR is spirV. This option is exclusive with -spirv_input. This option is exclusive with -llvm_input. -llvm_input Indicates that input file is an llvm binary. Default is OpenCL C kernel language. This option is exclusive with -spirv_input. This option is exclusive with -llvm_text. -spirv_input Indicates that input file is a spirV binary. Default is OpenCL C kernel language format. This option is exclusive with -llvm_input. This option is exclusive with -llvm_text. -options_name Will add suffix to output files. This suffix will be generated based on input options (useful when rebuilding with different set of options so that results won't get overwritten). This suffix is added always as the last part of the filename (even after file's extension). It does not affect '--output' parameter and can be used along with it ('--output' parameter defines the base name - i.e. prefix). -force_stos_opt Will forcibly enable stateless to stateful optimization, i.e. skip "-cl-intel-greater-than-4GB-buffer-required". -q Will silence output messages (except errors). -qq Will silence most of output messages. -spv_only Will generate only spirV file. -cpp_file Will generate c++ file with C-array containing Intel Compute device binary. -gen_file Will generate gen file. -output_no_suffix Prevents ocloc from adding family name suffix. --help Print this usage message. -revision_id Target stepping. Can be decimal or hexadecimal value. -exclude_ir Excludes IR from the output binary file. --format Enforce given binary format. The possible values are: --format zebin - Enforce generating zebin binary --format patchtokens - Enforce generating patchtokens (legacy) binary. -config Target hardware info config for a single device, e.g 1x4x8. Examples : Compile file to Intel Compute GPU device binary (out = source_file_Gen9core.bin) ocloc -file source_file.cl -device skl )===", getSupportedDevices(argHelper).c_str(), getDeprecatedDevices(argHelper).c_str()); } void OfflineCompiler::storeBinary( char *&pDst, size_t &dstSize, const void *pSrc, const size_t srcSize) { dstSize = 0; DEBUG_BREAK_IF(!(pSrc && srcSize > 0)); delete[] pDst; pDst = new char[srcSize]; dstSize = static_cast(srcSize); memcpy_s(pDst, dstSize, pSrc, srcSize); } bool OfflineCompiler::generateElfBinary() { if (!genBinary || !genBinarySize) { return false; } // return "as is" if zebin format if (isDeviceBinaryFormat(ArrayRef(reinterpret_cast(genBinary), genBinarySize))) { this->elfBinary = std::vector(genBinary, genBinary + genBinarySize); return true; } if (allowCaching) { elfHash = cache->getCachedFileName(getHardwareInfo(), genHash, options, internalOptions); auto loadedData = cache->loadCachedBinary(elfHash, elfBinarySize); elfBinary.assign(loadedData.get(), loadedData.get() + elfBinarySize); if (!elfBinary.empty()) { return true; } } SingleDeviceBinary binary = {}; binary.buildOptions = this->options; binary.intermediateRepresentation = ArrayRef(reinterpret_cast(this->irBinary), this->irBinarySize); binary.deviceBinary = ArrayRef(reinterpret_cast(this->genBinary), this->genBinarySize); binary.debugData = ArrayRef(reinterpret_cast(this->debugDataBinary), this->debugDataBinarySize); std::string packErrors; std::string packWarnings; using namespace NEO::Elf; ElfEncoder elfEncoder; elfEncoder.getElfFileHeader().type = ET_OPENCL_EXECUTABLE; if (binary.buildOptions.empty() == false) { elfEncoder.appendSection(SHT_OPENCL_OPTIONS, SectionNamesOpenCl::buildOptions, ArrayRef(reinterpret_cast(binary.buildOptions.data()), binary.buildOptions.size())); } if (!binary.intermediateRepresentation.empty() && !excludeIr) { if (isSpirV) { elfEncoder.appendSection(SHT_OPENCL_SPIRV, SectionNamesOpenCl::spirvObject, binary.intermediateRepresentation); } else { elfEncoder.appendSection(SHT_OPENCL_LLVM_BINARY, SectionNamesOpenCl::llvmObject, binary.intermediateRepresentation); } } if (binary.debugData.empty() == false) { elfEncoder.appendSection(SHT_OPENCL_DEV_DEBUG, SectionNamesOpenCl::deviceDebug, binary.debugData); } if (binary.deviceBinary.empty() == false) { elfEncoder.appendSection(SHT_OPENCL_DEV_BINARY, SectionNamesOpenCl::deviceBinary, binary.deviceBinary); } this->elfBinary = elfEncoder.encode(); if (allowCaching) { cache->cacheBinary(elfHash, reinterpret_cast(elfBinary.data()), static_cast(this->elfBinary.size())); } return true; } void OfflineCompiler::writeOutAllFiles() { std::string fileBase; std::string fileTrunk = getFileNameTrunk(inputFile); if (outputNoSuffix) { if (outputFile.empty()) { fileBase = fileTrunk; } else { fileBase = outputFile; } } else { if (outputFile.empty()) { fileBase = fileTrunk + "_" + productFamilyName; } else { fileBase = outputFile + "_" + productFamilyName; } } if (outputDirectory != "") { std::list dirList; std::string tmp = outputDirectory; size_t pos = outputDirectory.size() + 1; do { dirList.push_back(tmp); pos = tmp.find_last_of("/\\", pos); tmp = tmp.substr(0, pos); } while (pos != std::string::npos && !tmp.empty()); while (!dirList.empty()) { createDir(dirList.back()); dirList.pop_back(); } } if (irBinary && !inputFileSpirV) { std::string irOutputFileName = generateFilePathForIr(fileBase) + generateOptsSuffix(); argHelper->saveOutput(irOutputFileName, irBinary, irBinarySize); } if (genBinary) { if (useGenFile) { std::string genOutputFile = generateFilePath(outputDirectory, fileBase, ".gen") + generateOptsSuffix(); argHelper->saveOutput(genOutputFile, genBinary, genBinarySize); } if (useCppFile) { std::string cppOutputFile = generateFilePath(outputDirectory, fileBase, ".cpp"); std::string cpp = parseBinAsCharArray((uint8_t *)genBinary, genBinarySize, fileTrunk); argHelper->saveOutput(cppOutputFile, cpp.c_str(), cpp.size()); } } if (!elfBinary.empty()) { std::string elfOutputFile; if (outputNoSuffix) { elfOutputFile = generateFilePath(outputDirectory, fileBase, ""); } else { elfOutputFile = generateFilePath(outputDirectory, fileBase, ".bin") + generateOptsSuffix(); } argHelper->saveOutput( elfOutputFile, elfBinary.data(), elfBinary.size()); } if (debugDataBinary) { std::string debugOutputFile = generateFilePath(outputDirectory, fileBase, ".dbg") + generateOptsSuffix(); argHelper->saveOutput( debugOutputFile, debugDataBinary, debugDataBinarySize); } } void OfflineCompiler::createDir(const std::string &path) { MakeDirectory(path.c_str()); } bool OfflineCompiler::readOptionsFromFile(std::string &options, const std::string &file, OclocArgHelper *helper) { if (!helper->fileExists(file)) { return false; } size_t optionsSize = 0U; auto optionsFromFile = helper->loadDataFromFile(file, optionsSize); if (optionsSize > 0) { // Remove comment containing copyright header options = optionsFromFile.get(); size_t commentBegin = options.find("/*"); size_t commentEnd = options.rfind("*/"); if (commentBegin != std::string::npos && commentEnd != std::string::npos) { auto sizeToReplace = commentEnd - commentBegin + 2; options = options.replace(commentBegin, sizeToReplace, ""); size_t optionsBegin = options.find_first_not_of(" \t\n\r"); if (optionsBegin != std::string::npos) { options = options.substr(optionsBegin, options.length()); } } auto trimPos = options.find_last_not_of(" \n\r"); options = options.substr(0, trimPos + 1); } return true; } void OfflineCompiler::enforceFormat(std::string &format) { std::transform(format.begin(), format.end(), format.begin(), [](auto c) { return std::tolower(c); }); if (format == "zebin") { CompilerOptions::concatenateAppend(internalOptions, CompilerOptions::allowZebin); } else if (format == "patchtokens") { CompilerOptions::concatenateAppend(internalOptions, CompilerOptions::disableZebin); } else { argHelper->printf("Invalid format passed: %s. Ignoring.\n", format.c_str()); } } std::string generateFilePath(const std::string &directory, const std::string &fileNameBase, const char *extension) { UNRECOVERABLE_IF(extension == nullptr); if (directory.empty()) { return fileNameBase + extension; } bool hasTrailingSlash = (*directory.rbegin() == '/'); std::string ret; ret.reserve(directory.size() + (hasTrailingSlash ? 0 : 1) + fileNameBase.size() + strlen(extension) + 1); ret.append(directory); if (false == hasTrailingSlash) { ret.append("/", 1); } ret.append(fileNameBase); ret.append(extension); return ret; } } // namespace NEO