Main.cpp revision 6da4e253a513feef3405759fef6d0760828808ca
1d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao/* 2d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao * Copyright 2012, The Android Open Source Project 3d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao * 4d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao * Licensed under the Apache License, Version 2.0 (the "License"); 5d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao * you may not use this file except in compliance with the License. 6d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao * You may obtain a copy of the License at 7d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao * 8d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao * http://www.apache.org/licenses/LICENSE-2.0 9d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao * 10d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao * Unless required by applicable law or agreed to in writing, software 11d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao * distributed under the License is distributed on an "AS IS" BASIS, 12d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao * See the License for the specific language governing permissions and 14d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao * limitations under the License. 15d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao */ 16d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 170ab50b835805c75ad164466767c2c212f48954ccYang Ni#include <iostream> 180ab50b835805c75ad164466767c2c212f48954ccYang Ni#include <list> 190ab50b835805c75ad164466767c2c212f48954ccYang Ni#include <map> 200ab50b835805c75ad164466767c2c212f48954ccYang Ni#include <sstream> 21d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao#include <string> 22d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao#include <vector> 23d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 24c3437f05c638f8befda59170ae788873db24dc1cStephen Hines#include <dlfcn.h> 25d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao#include <stdlib.h> 26d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 27d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao#include <llvm/ADT/STLExtras.h> 28d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao#include <llvm/ADT/SmallString.h> 29d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao#include <llvm/Config/config.h> 30d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao#include <llvm/Support/CommandLine.h> 31d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao#include <llvm/Support/FileSystem.h> 32c46a3f5370364ad5698756d5da5b8845be9970d0Tim Murray#include <llvm/Support/ManagedStatic.h> 338be8dba08c97dff239ac8c6bdc7f3a52d96116aeStephen Hines#include <llvm/Support/MemoryBuffer.h> 341e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar#include <llvm/Support/Path.h> 35c3437f05c638f8befda59170ae788873db24dc1cStephen Hines#include <llvm/Support/PluginLoader.h> 36d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao#include <llvm/Support/raw_ostream.h> 37d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 38d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao#include <bcc/BCCContext.h> 39d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao#include <bcc/Compiler.h> 40d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao#include <bcc/Config/Config.h> 418be8dba08c97dff239ac8c6bdc7f3a52d96116aeStephen Hines#include <bcc/Renderscript/RSCompilerDriver.h> 42d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao#include <bcc/Script.h> 43d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao#include <bcc/Source.h> 44d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao#include <bcc/Support/CompilerConfig.h> 45d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao#include <bcc/Support/Initialization.h> 46d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao#include <bcc/Support/InputFile.h> 47d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao#include <bcc/Support/OutputFile.h> 48d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 49d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liaousing namespace bcc; 50d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 51c3437f05c638f8befda59170ae788873db24dc1cStephen Hines#define STR2(a) #a 52c3437f05c638f8befda59170ae788873db24dc1cStephen Hines#define STR(a) STR2(a) 53c3437f05c638f8befda59170ae788873db24dc1cStephen Hines 54d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao//===----------------------------------------------------------------------===// 55d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao// General Options 56d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao//===----------------------------------------------------------------------===// 57d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liaonamespace { 58d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 59a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Nillvm::cl::list<std::string> 60a4ded1373d7ad3e503f186e65bccf94126a0f020Yang NiOptInputFilenames(llvm::cl::Positional, llvm::cl::OneOrMore, 61a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni llvm::cl::desc("<input bitcode files>")); 62a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni 630ab50b835805c75ad164466767c2c212f48954ccYang Nillvm::cl::list<std::string> 640ab50b835805c75ad164466767c2c212f48954ccYang NiOptMergePlans("merge", llvm::cl::ZeroOrMore, 650ab50b835805c75ad164466767c2c212f48954ccYang Ni llvm::cl::desc("Lists of kernels to merge (as source-and-slot " 660ab50b835805c75ad164466767c2c212f48954ccYang Ni "pairs) and names for the final merged kernels")); 670ab50b835805c75ad164466767c2c212f48954ccYang Ni 680ab50b835805c75ad164466767c2c212f48954ccYang Nillvm::cl::list<std::string> 690ab50b835805c75ad164466767c2c212f48954ccYang NiOptInvokes("invoke", llvm::cl::ZeroOrMore, 700ab50b835805c75ad164466767c2c212f48954ccYang Ni llvm::cl::desc("Invocable functions")); 71d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 72d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liaollvm::cl::opt<std::string> 73d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei LiaoOptOutputFilename("o", llvm::cl::desc("Specify the output filename"), 748be8dba08c97dff239ac8c6bdc7f3a52d96116aeStephen Hines llvm::cl::value_desc("filename"), 758be8dba08c97dff239ac8c6bdc7f3a52d96116aeStephen Hines llvm::cl::init("bcc_output")); 768be8dba08c97dff239ac8c6bdc7f3a52d96116aeStephen Hines 778be8dba08c97dff239ac8c6bdc7f3a52d96116aeStephen Hinesllvm::cl::opt<std::string> 788be8dba08c97dff239ac8c6bdc7f3a52d96116aeStephen HinesOptBCLibFilename("bclib", llvm::cl::desc("Specify the bclib filename"), 798be8dba08c97dff239ac8c6bdc7f3a52d96116aeStephen Hines llvm::cl::value_desc("bclib")); 808be8dba08c97dff239ac8c6bdc7f3a52d96116aeStephen Hines 818be8dba08c97dff239ac8c6bdc7f3a52d96116aeStephen Hinesllvm::cl::opt<std::string> 826da4e253a513feef3405759fef6d0760828808caYang NiOptBCLibRelaxedFilename("bclib_relaxed", llvm::cl::desc("Specify the bclib filename optimized for " 836da4e253a513feef3405759fef6d0760828808caYang Ni "relaxed precision floating point maths"), 846da4e253a513feef3405759fef6d0760828808caYang Ni llvm::cl::init(""), 856da4e253a513feef3405759fef6d0760828808caYang Ni llvm::cl::value_desc("bclib_relaxed")); 866da4e253a513feef3405759fef6d0760828808caYang Ni 876da4e253a513feef3405759fef6d0760828808caYang Nillvm::cl::opt<std::string> 888be8dba08c97dff239ac8c6bdc7f3a52d96116aeStephen HinesOptOutputPath("output_path", llvm::cl::desc("Specify the output path"), 898be8dba08c97dff239ac8c6bdc7f3a52d96116aeStephen Hines llvm::cl::value_desc("output path"), 908be8dba08c97dff239ac8c6bdc7f3a52d96116aeStephen Hines llvm::cl::init(".")); 91d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 927b980e1717f3cf418f7bc4e40597004bc1139b8bTobias Grosserllvm::cl::opt<bool> 937b980e1717f3cf418f7bc4e40597004bc1139b8bTobias GrosserOptEmitLLVM("emit-llvm", 947b980e1717f3cf418f7bc4e40597004bc1139b8bTobias Grosser llvm::cl::desc("Emit an LLVM-IR version of the generated program")); 957b980e1717f3cf418f7bc4e40597004bc1139b8bTobias Grosser 96d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liaollvm::cl::opt<std::string> 97d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei LiaoOptTargetTriple("mtriple", 98d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao llvm::cl::desc("Specify the target triple (default: " 99d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao DEFAULT_TARGET_TRIPLE_STRING ")"), 100d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao llvm::cl::init(DEFAULT_TARGET_TRIPLE_STRING), 101d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao llvm::cl::value_desc("triple")); 102d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 103d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liaollvm::cl::alias OptTargetTripleC("C", llvm::cl::NotHidden, 104d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao llvm::cl::desc("Alias for -mtriple"), 105d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao llvm::cl::aliasopt(OptTargetTriple)); 106d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 107c3437f05c638f8befda59170ae788873db24dc1cStephen Hinesllvm::cl::opt<bool> 108c3437f05c638f8befda59170ae788873db24dc1cStephen HinesOptRSDebugContext("rs-debug-ctx", 109c3437f05c638f8befda59170ae788873db24dc1cStephen Hines llvm::cl::desc("Enable build to work with a RenderScript debug context")); 110c3437f05c638f8befda59170ae788873db24dc1cStephen Hines 11151ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga Nainarllvm::cl::opt<std::string> 11251ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga NainarOptChecksum("build-checksum", 11351ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga Nainar llvm::cl::desc("Embed a checksum of this compiler invocation for" 11451ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga Nainar " cache invalidation at a later time"), 11551ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga Nainar llvm::cl::value_desc("checksum")); 11651ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga Nainar 117d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao//===----------------------------------------------------------------------===// 118d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao// Compiler Options 119d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao//===----------------------------------------------------------------------===// 1201e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainarllvm::cl::opt<bool> 1211e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga NainarOptPIC("fPIC", llvm::cl::desc("Generate fully relocatable, position independent" 1221e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar " code")); 1231e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar 1241e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar// If set, use buildForCompatLib to embed RS symbol information into the object 1251e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar// file. The information is stored in the .rs.info variable. This option is 1261e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar// to be used in tandem with -fPIC. 1271e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainarllvm::cl::opt<bool> 1281e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga NainarOptEmbedRSInfo("embedRSInfo", 1291e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar llvm::cl::desc("Embed RS Info into the object file instead of generating" 1301e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar " a separate .o.info file")); 131d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 1327e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hines// RenderScript uses -O3 by default 133d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liaollvm::cl::opt<char> 134d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei LiaoOptOptLevel("O", llvm::cl::desc("Optimization level. [-O0, -O1, -O2, or -O3] " 1357e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hines "(default: -O3)"), 1367e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hines llvm::cl::Prefix, llvm::cl::ZeroOrMore, llvm::cl::init('3')); 137d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 138d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao// Override "bcc -version" since the LLVM version information is not correct on 139d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao// Android build. 140d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liaovoid BCCVersionPrinter() { 141d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao llvm::raw_ostream &os = llvm::outs(); 142d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao os << "libbcc (The Android Open Source Project, http://www.android.com/):\n" 143c5e607adff80a66bc5420baffd299862abdf368dJean-Luc Brouillet << " Default target: " << DEFAULT_TARGET_TRIPLE_STRING << "\n\n" 144c5e607adff80a66bc5420baffd299862abdf368dJean-Luc Brouillet << "LLVM (http://llvm.org/):\n" 145d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao << " Version: " << PACKAGE_VERSION << "\n"; 146d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao return; 147d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao} 148d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 1490ab50b835805c75ad164466767c2c212f48954ccYang Nivoid extractSourcesAndSlots(const llvm::cl::list<std::string>& optList, 1500ab50b835805c75ad164466767c2c212f48954ccYang Ni std::list<std::string>* batchNames, 1510ab50b835805c75ad164466767c2c212f48954ccYang Ni std::list<std::list<std::pair<int, int>>>* sourcesAndSlots) { 1520ab50b835805c75ad164466767c2c212f48954ccYang Ni for (unsigned i = 0; i < optList.size(); ++i) { 1530ab50b835805c75ad164466767c2c212f48954ccYang Ni std::string plan = optList[i]; 1540ab50b835805c75ad164466767c2c212f48954ccYang Ni unsigned found = plan.find(":"); 1550ab50b835805c75ad164466767c2c212f48954ccYang Ni 1560ab50b835805c75ad164466767c2c212f48954ccYang Ni std::string name = plan.substr(0, found); 1570ab50b835805c75ad164466767c2c212f48954ccYang Ni std::cerr << "new kernel name: " << name << std::endl; 1580ab50b835805c75ad164466767c2c212f48954ccYang Ni batchNames->push_back(name); 1590ab50b835805c75ad164466767c2c212f48954ccYang Ni 1600ab50b835805c75ad164466767c2c212f48954ccYang Ni std::istringstream iss(plan.substr(found + 1)); 1610ab50b835805c75ad164466767c2c212f48954ccYang Ni std::string s; 1620ab50b835805c75ad164466767c2c212f48954ccYang Ni std::list<std::pair<int, int>> planList; 1630ab50b835805c75ad164466767c2c212f48954ccYang Ni while (getline(iss, s, '.')) { 1640ab50b835805c75ad164466767c2c212f48954ccYang Ni found = s.find(","); 1650ab50b835805c75ad164466767c2c212f48954ccYang Ni std::string sourceStr = s.substr(0, found); 1660ab50b835805c75ad164466767c2c212f48954ccYang Ni std::string slotStr = s.substr(found + 1); 1670ab50b835805c75ad164466767c2c212f48954ccYang Ni 1680ab50b835805c75ad164466767c2c212f48954ccYang Ni std::cerr << "source " << sourceStr << ", slot " << slotStr << std::endl; 1690ab50b835805c75ad164466767c2c212f48954ccYang Ni 1700ab50b835805c75ad164466767c2c212f48954ccYang Ni int source = std::stoi(sourceStr); 1710ab50b835805c75ad164466767c2c212f48954ccYang Ni int slot = std::stoi(slotStr); 1720ab50b835805c75ad164466767c2c212f48954ccYang Ni planList.push_back(std::make_pair(source, slot)); 1730ab50b835805c75ad164466767c2c212f48954ccYang Ni } 174a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni 1750ab50b835805c75ad164466767c2c212f48954ccYang Ni sourcesAndSlots->push_back(planList); 1760ab50b835805c75ad164466767c2c212f48954ccYang Ni } 1770ab50b835805c75ad164466767c2c212f48954ccYang Ni} 178a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni 1790ab50b835805c75ad164466767c2c212f48954ccYang Nibool compileScriptGroup(BCCContext& Context, RSCompilerDriver& RSCD) { 1800ab50b835805c75ad164466767c2c212f48954ccYang Ni std::vector<bcc::Source*> sources; 181a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni for (unsigned i = 0; i < OptInputFilenames.size(); ++i) { 1820ab50b835805c75ad164466767c2c212f48954ccYang Ni bcc::Source* source = 183a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni bcc::Source::CreateFromFile(Context, OptInputFilenames[i]); 184a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni if (!source) { 185a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni llvm::errs() << "Error loading file '" << OptInputFilenames[i]<< "'\n"; 186a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni return false; 187a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni } 188a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni sources.push_back(source); 189a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni } 190a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni 1910ab50b835805c75ad164466767c2c212f48954ccYang Ni std::list<std::string> fusedKernelNames; 1920ab50b835805c75ad164466767c2c212f48954ccYang Ni std::list<std::list<std::pair<int, int>>> sourcesAndSlots; 1930ab50b835805c75ad164466767c2c212f48954ccYang Ni extractSourcesAndSlots(OptMergePlans, &fusedKernelNames, &sourcesAndSlots); 1940ab50b835805c75ad164466767c2c212f48954ccYang Ni 1950ab50b835805c75ad164466767c2c212f48954ccYang Ni std::list<std::string> invokeBatchNames; 1960ab50b835805c75ad164466767c2c212f48954ccYang Ni std::list<std::list<std::pair<int, int>>> invokeSourcesAndSlots; 1970ab50b835805c75ad164466767c2c212f48954ccYang Ni extractSourcesAndSlots(OptInvokes, &invokeBatchNames, &invokeSourcesAndSlots); 1980ab50b835805c75ad164466767c2c212f48954ccYang Ni 199a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni std::string outputFilepath(OptOutputPath); 200a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni outputFilepath.append("/"); 201a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni outputFilepath.append(OptOutputFilename); 202a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni 203a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni bool success = RSCD.buildScriptGroup( 2046da4e253a513feef3405759fef6d0760828808caYang Ni Context, outputFilepath.c_str(), OptBCLibFilename.c_str(), 2056da4e253a513feef3405759fef6d0760828808caYang Ni OptBCLibRelaxedFilename.c_str(), true, 2060ab50b835805c75ad164466767c2c212f48954ccYang Ni sources, sourcesAndSlots, fusedKernelNames, 2070ab50b835805c75ad164466767c2c212f48954ccYang Ni invokeSourcesAndSlots, invokeBatchNames); 208a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni 209a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni return success; 210a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni} 211a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni 212d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao} // end anonymous namespace 213d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 214d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liaostatic inline 2157e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hinesbool ConfigCompiler(RSCompilerDriver &pRSCD) { 216b4447cd2b14f53efd9102d28da48000be7b2d4fdChris Wailes Compiler *RSC = pRSCD.getCompiler(); 217900c6c1f08f7c572125d7d39abe0f0f9eafbfa14Chris Wailes CompilerConfig *config = nullptr; 218d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 219d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao config = new (std::nothrow) CompilerConfig(OptTargetTriple); 220900c6c1f08f7c572125d7d39abe0f0f9eafbfa14Chris Wailes if (config == nullptr) { 221d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao llvm::errs() << "Out of memory when create the compiler configuration!\n"; 222d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao return false; 223d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao } 224d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 225602ce0336c0ff5109923cee9b4426e913e718605Jian Li // llvm3.5 has removed the auto-detect feature for x86 subtarget, 226602ce0336c0ff5109923cee9b4426e913e718605Jian Li // so set features explicitly in bcc. 227602ce0336c0ff5109923cee9b4426e913e718605Jian Li if ((config->getTriple().find("i686") != std::string::npos) || 228602ce0336c0ff5109923cee9b4426e913e718605Jian Li (config->getTriple().find("x86_64") != std::string::npos)) { 229602ce0336c0ff5109923cee9b4426e913e718605Jian Li std::vector<std::string> fv; 230602ce0336c0ff5109923cee9b4426e913e718605Jian Li 231602ce0336c0ff5109923cee9b4426e913e718605Jian Li#if defined(__SSE3__) 232602ce0336c0ff5109923cee9b4426e913e718605Jian Li fv.push_back("+sse3"); 233602ce0336c0ff5109923cee9b4426e913e718605Jian Li#endif 234602ce0336c0ff5109923cee9b4426e913e718605Jian Li#if defined(__SSSE3__) 235602ce0336c0ff5109923cee9b4426e913e718605Jian Li fv.push_back("+ssse3"); 236602ce0336c0ff5109923cee9b4426e913e718605Jian Li#endif 237602ce0336c0ff5109923cee9b4426e913e718605Jian Li#if defined(__SSE4_1__) 238602ce0336c0ff5109923cee9b4426e913e718605Jian Li fv.push_back("+sse4.1"); 239602ce0336c0ff5109923cee9b4426e913e718605Jian Li#endif 240602ce0336c0ff5109923cee9b4426e913e718605Jian Li#if defined(__SSE4_2__) 241602ce0336c0ff5109923cee9b4426e913e718605Jian Li fv.push_back("+sse4.2"); 242602ce0336c0ff5109923cee9b4426e913e718605Jian Li#endif 243602ce0336c0ff5109923cee9b4426e913e718605Jian Li 244602ce0336c0ff5109923cee9b4426e913e718605Jian Li if (fv.size()) { 245602ce0336c0ff5109923cee9b4426e913e718605Jian Li config->setFeatureString(fv); 246602ce0336c0ff5109923cee9b4426e913e718605Jian Li } 247602ce0336c0ff5109923cee9b4426e913e718605Jian Li } 248602ce0336c0ff5109923cee9b4426e913e718605Jian Li 2491e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar if (OptPIC) { 2501e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar config->setRelocationModel(llvm::Reloc::PIC_); 2511e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar 2521e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar // For x86_64, CodeModel needs to be small if PIC_ reloc is used. 2531e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar // Otherwise, we end up with TEXTRELs in the shared library. 2541e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar if (config->getTriple().find("x86_64") != std::string::npos) { 2551e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar config->setCodeModel(llvm::CodeModel::Small); 2561e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar } 2571e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar } 258d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao switch (OptOptLevel) { 259d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao case '0': config->setOptimizationLevel(llvm::CodeGenOpt::None); break; 260d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao case '1': config->setOptimizationLevel(llvm::CodeGenOpt::Less); break; 2617e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hines case '2': config->setOptimizationLevel(llvm::CodeGenOpt::Default); break; 2627e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hines case '3': 263d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao default: { 2647e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hines config->setOptimizationLevel(llvm::CodeGenOpt::Aggressive); 265d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao break; 266d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao } 267d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao } 268d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 2697e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hines pRSCD.setConfig(config); 2707e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hines Compiler::ErrorCode result = RSC->config(*config); 271d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 272c3437f05c638f8befda59170ae788873db24dc1cStephen Hines if (OptRSDebugContext) { 273c3437f05c638f8befda59170ae788873db24dc1cStephen Hines pRSCD.setDebugContext(true); 274c3437f05c638f8befda59170ae788873db24dc1cStephen Hines } 275c3437f05c638f8befda59170ae788873db24dc1cStephen Hines 276d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao if (result != Compiler::kSuccess) { 277d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao llvm::errs() << "Failed to configure the compiler! (detail: " 278d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao << Compiler::GetErrorString(result) << ")\n"; 279d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao return false; 280d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao } 281d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 282d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao return true; 283d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao} 284d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 285d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liaoint main(int argc, char **argv) { 286c46a3f5370364ad5698756d5da5b8845be9970d0Tim Murray 287c46a3f5370364ad5698756d5da5b8845be9970d0Tim Murray llvm::llvm_shutdown_obj Y; 288c46a3f5370364ad5698756d5da5b8845be9970d0Tim Murray init::Initialize(); 289d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao llvm::cl::SetVersionPrinter(BCCVersionPrinter); 290d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao llvm::cl::ParseCommandLineOptions(argc, argv); 291f2ac3176c351cd80bce77fe1488f3de2d0789c1bJean-Luc Brouillet std::string commandLine = bcc::getCommandLine(argc, argv); 292d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 293d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao BCCContext context; 2948be8dba08c97dff239ac8c6bdc7f3a52d96116aeStephen Hines RSCompilerDriver RSCD; 295d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 296c5e607adff80a66bc5420baffd299862abdf368dJean-Luc Brouillet if (OptBCLibFilename.empty()) { 297a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni ALOGE("Failed to compile bitcode, -bclib was not specified"); 298d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao return EXIT_FAILURE; 299d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao } 3008be8dba08c97dff239ac8c6bdc7f3a52d96116aeStephen Hines 3017e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hines if (!ConfigCompiler(RSCD)) { 3027e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hines ALOGE("Failed to configure compiler"); 3037e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hines return EXIT_FAILURE; 3047e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hines } 305c3437f05c638f8befda59170ae788873db24dc1cStephen Hines 306c3437f05c638f8befda59170ae788873db24dc1cStephen Hines // Attempt to dynamically initialize the compiler driver if such a function 307c3437f05c638f8befda59170ae788873db24dc1cStephen Hines // is present. It is only present if passed via "-load libFOO.so". 308c3437f05c638f8befda59170ae788873db24dc1cStephen Hines RSCompilerDriverInit_t rscdi = (RSCompilerDriverInit_t) 309c3437f05c638f8befda59170ae788873db24dc1cStephen Hines dlsym(RTLD_DEFAULT, STR(RS_COMPILER_DRIVER_INIT_FN)); 310900c6c1f08f7c572125d7d39abe0f0f9eafbfa14Chris Wailes if (rscdi != nullptr) { 311c3437f05c638f8befda59170ae788873db24dc1cStephen Hines rscdi(&RSCD); 312c3437f05c638f8befda59170ae788873db24dc1cStephen Hines } 313c3437f05c638f8befda59170ae788873db24dc1cStephen Hines 3140ab50b835805c75ad164466767c2c212f48954ccYang Ni if (OptMergePlans.size() > 0) { 3150ab50b835805c75ad164466767c2c212f48954ccYang Ni bool success = compileScriptGroup(context, RSCD); 3160ab50b835805c75ad164466767c2c212f48954ccYang Ni 317a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni if (!success) { 318a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni return EXIT_FAILURE; 319a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni } 3200ab50b835805c75ad164466767c2c212f48954ccYang Ni 321a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni return EXIT_SUCCESS; 322a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni } 323a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni 324a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni llvm::ErrorOr<std::unique_ptr<llvm::MemoryBuffer>> mb_or_error = 325a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni llvm::MemoryBuffer::getFile(OptInputFilenames[0].c_str()); 326a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni if (mb_or_error.getError()) { 327a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni ALOGE("Failed to load bitcode from path %s! (%s)", 328a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni OptInputFilenames[0].c_str(), mb_or_error.getError().message().c_str()); 329a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni return EXIT_FAILURE; 330a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni } 331a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni std::unique_ptr<llvm::MemoryBuffer> input_data = std::move(mb_or_error.get()); 332a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni 333a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni const char *bitcode = input_data->getBufferStart(); 334a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni size_t bitcodeSize = input_data->getBufferSize(); 335a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni 3361e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar if (!OptEmbedRSInfo) { 3371e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar bool built = RSCD.build(context, OptOutputPath.c_str(), OptOutputFilename.c_str(), bitcode, 33851ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga Nainar bitcodeSize, commandLine.c_str(), OptChecksum.c_str(), OptBCLibFilename.c_str(), nullptr, 3391e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar OptEmitLLVM); 340d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 3411e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar if (!built) { 3421e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar return EXIT_FAILURE; 3431e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar } 3440ab50b835805c75ad164466767c2c212f48954ccYang Ni } else { 3451e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar // embedRSInfo is set. Use buildForCompatLib to embed RS symbol information 3461e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar // into the .rs.info symbol. 347a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni Source *source = Source::CreateFromBuffer(context, OptInputFilenames[0].c_str(), 3481e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar bitcode, bitcodeSize); 3491e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar RSScript *s = new (std::nothrow) RSScript(*source); 3501e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar if (s == nullptr) { 3511e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar llvm::errs() << "Out of memory when creating script for file `" 352a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni << OptInputFilenames[0] << "'!\n"; 3531e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar delete source; 3541e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar return EXIT_FAILURE; 3551e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar } 3561e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar 3571e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar llvm::SmallString<80> output(OptOutputPath); 3581e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar llvm::sys::path::append(output, "/", OptOutputFilename); 3591e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar llvm::sys::path::replace_extension(output, ".o"); 3601e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar 36151ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga Nainar if (!RSCD.buildForCompatLib(*s, output.c_str(), OptChecksum.c_str(), 36251ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga Nainar OptBCLibFilename.c_str(), OptEmitLLVM)) { 3631e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar fprintf(stderr, "Failed to compile script!"); 3641e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar return EXIT_FAILURE; 3651e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar } 366d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao } 367d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 368d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao return EXIT_SUCCESS; 369d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao} 370