Main.cpp revision 0ab50b835805c75ad164466767c2c212f48954cc
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> 828be8dba08c97dff239ac8c6bdc7f3a52d96116aeStephen HinesOptOutputPath("output_path", llvm::cl::desc("Specify the output path"), 838be8dba08c97dff239ac8c6bdc7f3a52d96116aeStephen Hines llvm::cl::value_desc("output path"), 848be8dba08c97dff239ac8c6bdc7f3a52d96116aeStephen Hines llvm::cl::init(".")); 85d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 867b980e1717f3cf418f7bc4e40597004bc1139b8bTobias Grosserllvm::cl::opt<bool> 877b980e1717f3cf418f7bc4e40597004bc1139b8bTobias GrosserOptEmitLLVM("emit-llvm", 887b980e1717f3cf418f7bc4e40597004bc1139b8bTobias Grosser llvm::cl::desc("Emit an LLVM-IR version of the generated program")); 897b980e1717f3cf418f7bc4e40597004bc1139b8bTobias Grosser 90d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liaollvm::cl::opt<std::string> 91d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei LiaoOptTargetTriple("mtriple", 92d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao llvm::cl::desc("Specify the target triple (default: " 93d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao DEFAULT_TARGET_TRIPLE_STRING ")"), 94d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao llvm::cl::init(DEFAULT_TARGET_TRIPLE_STRING), 95d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao llvm::cl::value_desc("triple")); 96d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 97d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liaollvm::cl::alias OptTargetTripleC("C", llvm::cl::NotHidden, 98d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao llvm::cl::desc("Alias for -mtriple"), 99d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao llvm::cl::aliasopt(OptTargetTriple)); 100d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 101c3437f05c638f8befda59170ae788873db24dc1cStephen Hinesllvm::cl::opt<bool> 102c3437f05c638f8befda59170ae788873db24dc1cStephen HinesOptRSDebugContext("rs-debug-ctx", 103c3437f05c638f8befda59170ae788873db24dc1cStephen Hines llvm::cl::desc("Enable build to work with a RenderScript debug context")); 104c3437f05c638f8befda59170ae788873db24dc1cStephen Hines 10551ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga Nainarllvm::cl::opt<std::string> 10651ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga NainarOptChecksum("build-checksum", 10751ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga Nainar llvm::cl::desc("Embed a checksum of this compiler invocation for" 10851ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga Nainar " cache invalidation at a later time"), 10951ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga Nainar llvm::cl::value_desc("checksum")); 11051ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga Nainar 111d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao//===----------------------------------------------------------------------===// 112d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao// Compiler Options 113d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao//===----------------------------------------------------------------------===// 1141e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainarllvm::cl::opt<bool> 1151e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga NainarOptPIC("fPIC", llvm::cl::desc("Generate fully relocatable, position independent" 1161e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar " code")); 1171e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar 1181e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar// If set, use buildForCompatLib to embed RS symbol information into the object 1191e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar// file. The information is stored in the .rs.info variable. This option is 1201e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar// to be used in tandem with -fPIC. 1211e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainarllvm::cl::opt<bool> 1221e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga NainarOptEmbedRSInfo("embedRSInfo", 1231e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar llvm::cl::desc("Embed RS Info into the object file instead of generating" 1241e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar " a separate .o.info file")); 125d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 1267e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hines// RenderScript uses -O3 by default 127d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liaollvm::cl::opt<char> 128d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei LiaoOptOptLevel("O", llvm::cl::desc("Optimization level. [-O0, -O1, -O2, or -O3] " 1297e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hines "(default: -O3)"), 1307e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hines llvm::cl::Prefix, llvm::cl::ZeroOrMore, llvm::cl::init('3')); 131d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 132d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao// Override "bcc -version" since the LLVM version information is not correct on 133d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao// Android build. 134d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liaovoid BCCVersionPrinter() { 135d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao llvm::raw_ostream &os = llvm::outs(); 136d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao os << "libbcc (The Android Open Source Project, http://www.android.com/):\n" 137c5e607adff80a66bc5420baffd299862abdf368dJean-Luc Brouillet << " Default target: " << DEFAULT_TARGET_TRIPLE_STRING << "\n\n" 138c5e607adff80a66bc5420baffd299862abdf368dJean-Luc Brouillet << "LLVM (http://llvm.org/):\n" 139d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao << " Version: " << PACKAGE_VERSION << "\n"; 140d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao return; 141d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao} 142d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 1430ab50b835805c75ad164466767c2c212f48954ccYang Nivoid extractSourcesAndSlots(const llvm::cl::list<std::string>& optList, 1440ab50b835805c75ad164466767c2c212f48954ccYang Ni std::list<std::string>* batchNames, 1450ab50b835805c75ad164466767c2c212f48954ccYang Ni std::list<std::list<std::pair<int, int>>>* sourcesAndSlots) { 1460ab50b835805c75ad164466767c2c212f48954ccYang Ni for (unsigned i = 0; i < optList.size(); ++i) { 1470ab50b835805c75ad164466767c2c212f48954ccYang Ni std::string plan = optList[i]; 1480ab50b835805c75ad164466767c2c212f48954ccYang Ni unsigned found = plan.find(":"); 1490ab50b835805c75ad164466767c2c212f48954ccYang Ni 1500ab50b835805c75ad164466767c2c212f48954ccYang Ni std::string name = plan.substr(0, found); 1510ab50b835805c75ad164466767c2c212f48954ccYang Ni std::cerr << "new kernel name: " << name << std::endl; 1520ab50b835805c75ad164466767c2c212f48954ccYang Ni batchNames->push_back(name); 1530ab50b835805c75ad164466767c2c212f48954ccYang Ni 1540ab50b835805c75ad164466767c2c212f48954ccYang Ni std::istringstream iss(plan.substr(found + 1)); 1550ab50b835805c75ad164466767c2c212f48954ccYang Ni std::string s; 1560ab50b835805c75ad164466767c2c212f48954ccYang Ni std::list<std::pair<int, int>> planList; 1570ab50b835805c75ad164466767c2c212f48954ccYang Ni while (getline(iss, s, '.')) { 1580ab50b835805c75ad164466767c2c212f48954ccYang Ni found = s.find(","); 1590ab50b835805c75ad164466767c2c212f48954ccYang Ni std::string sourceStr = s.substr(0, found); 1600ab50b835805c75ad164466767c2c212f48954ccYang Ni std::string slotStr = s.substr(found + 1); 1610ab50b835805c75ad164466767c2c212f48954ccYang Ni 1620ab50b835805c75ad164466767c2c212f48954ccYang Ni std::cerr << "source " << sourceStr << ", slot " << slotStr << std::endl; 1630ab50b835805c75ad164466767c2c212f48954ccYang Ni 1640ab50b835805c75ad164466767c2c212f48954ccYang Ni int source = std::stoi(sourceStr); 1650ab50b835805c75ad164466767c2c212f48954ccYang Ni int slot = std::stoi(slotStr); 1660ab50b835805c75ad164466767c2c212f48954ccYang Ni planList.push_back(std::make_pair(source, slot)); 1670ab50b835805c75ad164466767c2c212f48954ccYang Ni } 168a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni 1690ab50b835805c75ad164466767c2c212f48954ccYang Ni sourcesAndSlots->push_back(planList); 1700ab50b835805c75ad164466767c2c212f48954ccYang Ni } 1710ab50b835805c75ad164466767c2c212f48954ccYang Ni} 172a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni 1730ab50b835805c75ad164466767c2c212f48954ccYang Nibool compileScriptGroup(BCCContext& Context, RSCompilerDriver& RSCD) { 1740ab50b835805c75ad164466767c2c212f48954ccYang Ni std::vector<bcc::Source*> sources; 175a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni for (unsigned i = 0; i < OptInputFilenames.size(); ++i) { 1760ab50b835805c75ad164466767c2c212f48954ccYang Ni bcc::Source* source = 177a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni bcc::Source::CreateFromFile(Context, OptInputFilenames[i]); 178a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni if (!source) { 179a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni llvm::errs() << "Error loading file '" << OptInputFilenames[i]<< "'\n"; 180a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni return false; 181a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni } 182a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni sources.push_back(source); 183a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni } 184a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni 1850ab50b835805c75ad164466767c2c212f48954ccYang Ni std::list<std::string> fusedKernelNames; 1860ab50b835805c75ad164466767c2c212f48954ccYang Ni std::list<std::list<std::pair<int, int>>> sourcesAndSlots; 1870ab50b835805c75ad164466767c2c212f48954ccYang Ni extractSourcesAndSlots(OptMergePlans, &fusedKernelNames, &sourcesAndSlots); 1880ab50b835805c75ad164466767c2c212f48954ccYang Ni 1890ab50b835805c75ad164466767c2c212f48954ccYang Ni std::list<std::string> invokeBatchNames; 1900ab50b835805c75ad164466767c2c212f48954ccYang Ni std::list<std::list<std::pair<int, int>>> invokeSourcesAndSlots; 1910ab50b835805c75ad164466767c2c212f48954ccYang Ni extractSourcesAndSlots(OptInvokes, &invokeBatchNames, &invokeSourcesAndSlots); 1920ab50b835805c75ad164466767c2c212f48954ccYang Ni 193a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni std::string outputFilepath(OptOutputPath); 194a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni outputFilepath.append("/"); 195a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni outputFilepath.append(OptOutputFilename); 196a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni 197a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni bool success = RSCD.buildScriptGroup( 1980ab50b835805c75ad164466767c2c212f48954ccYang Ni Context, outputFilepath.c_str(), OptBCLibFilename.c_str(), true, 1990ab50b835805c75ad164466767c2c212f48954ccYang Ni sources, sourcesAndSlots, fusedKernelNames, 2000ab50b835805c75ad164466767c2c212f48954ccYang Ni invokeSourcesAndSlots, invokeBatchNames); 201a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni 202a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni return success; 203a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni} 204a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni 205d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao} // end anonymous namespace 206d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 207d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liaostatic inline 2087e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hinesbool ConfigCompiler(RSCompilerDriver &pRSCD) { 209b4447cd2b14f53efd9102d28da48000be7b2d4fdChris Wailes Compiler *RSC = pRSCD.getCompiler(); 210900c6c1f08f7c572125d7d39abe0f0f9eafbfa14Chris Wailes CompilerConfig *config = nullptr; 211d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 212d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao config = new (std::nothrow) CompilerConfig(OptTargetTriple); 213900c6c1f08f7c572125d7d39abe0f0f9eafbfa14Chris Wailes if (config == nullptr) { 214d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao llvm::errs() << "Out of memory when create the compiler configuration!\n"; 215d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao return false; 216d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao } 217d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 218602ce0336c0ff5109923cee9b4426e913e718605Jian Li // llvm3.5 has removed the auto-detect feature for x86 subtarget, 219602ce0336c0ff5109923cee9b4426e913e718605Jian Li // so set features explicitly in bcc. 220602ce0336c0ff5109923cee9b4426e913e718605Jian Li if ((config->getTriple().find("i686") != std::string::npos) || 221602ce0336c0ff5109923cee9b4426e913e718605Jian Li (config->getTriple().find("x86_64") != std::string::npos)) { 222602ce0336c0ff5109923cee9b4426e913e718605Jian Li std::vector<std::string> fv; 223602ce0336c0ff5109923cee9b4426e913e718605Jian Li 224602ce0336c0ff5109923cee9b4426e913e718605Jian Li#if defined(__SSE3__) 225602ce0336c0ff5109923cee9b4426e913e718605Jian Li fv.push_back("+sse3"); 226602ce0336c0ff5109923cee9b4426e913e718605Jian Li#endif 227602ce0336c0ff5109923cee9b4426e913e718605Jian Li#if defined(__SSSE3__) 228602ce0336c0ff5109923cee9b4426e913e718605Jian Li fv.push_back("+ssse3"); 229602ce0336c0ff5109923cee9b4426e913e718605Jian Li#endif 230602ce0336c0ff5109923cee9b4426e913e718605Jian Li#if defined(__SSE4_1__) 231602ce0336c0ff5109923cee9b4426e913e718605Jian Li fv.push_back("+sse4.1"); 232602ce0336c0ff5109923cee9b4426e913e718605Jian Li#endif 233602ce0336c0ff5109923cee9b4426e913e718605Jian Li#if defined(__SSE4_2__) 234602ce0336c0ff5109923cee9b4426e913e718605Jian Li fv.push_back("+sse4.2"); 235602ce0336c0ff5109923cee9b4426e913e718605Jian Li#endif 236602ce0336c0ff5109923cee9b4426e913e718605Jian Li 237602ce0336c0ff5109923cee9b4426e913e718605Jian Li if (fv.size()) { 238602ce0336c0ff5109923cee9b4426e913e718605Jian Li config->setFeatureString(fv); 239602ce0336c0ff5109923cee9b4426e913e718605Jian Li } 240602ce0336c0ff5109923cee9b4426e913e718605Jian Li } 241602ce0336c0ff5109923cee9b4426e913e718605Jian Li 2421e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar if (OptPIC) { 2431e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar config->setRelocationModel(llvm::Reloc::PIC_); 2441e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar 2451e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar // For x86_64, CodeModel needs to be small if PIC_ reloc is used. 2461e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar // Otherwise, we end up with TEXTRELs in the shared library. 2471e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar if (config->getTriple().find("x86_64") != std::string::npos) { 2481e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar config->setCodeModel(llvm::CodeModel::Small); 2491e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar } 2501e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar } 251d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao switch (OptOptLevel) { 252d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao case '0': config->setOptimizationLevel(llvm::CodeGenOpt::None); break; 253d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao case '1': config->setOptimizationLevel(llvm::CodeGenOpt::Less); break; 2547e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hines case '2': config->setOptimizationLevel(llvm::CodeGenOpt::Default); break; 2557e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hines case '3': 256d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao default: { 2577e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hines config->setOptimizationLevel(llvm::CodeGenOpt::Aggressive); 258d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao break; 259d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao } 260d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao } 261d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 2627e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hines pRSCD.setConfig(config); 2637e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hines Compiler::ErrorCode result = RSC->config(*config); 264d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 265c3437f05c638f8befda59170ae788873db24dc1cStephen Hines if (OptRSDebugContext) { 266c3437f05c638f8befda59170ae788873db24dc1cStephen Hines pRSCD.setDebugContext(true); 267c3437f05c638f8befda59170ae788873db24dc1cStephen Hines } 268c3437f05c638f8befda59170ae788873db24dc1cStephen Hines 269d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao if (result != Compiler::kSuccess) { 270d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao llvm::errs() << "Failed to configure the compiler! (detail: " 271d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao << Compiler::GetErrorString(result) << ")\n"; 272d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao return false; 273d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao } 274d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 275d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao return true; 276d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao} 277d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 278d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liaoint main(int argc, char **argv) { 279c46a3f5370364ad5698756d5da5b8845be9970d0Tim Murray 280c46a3f5370364ad5698756d5da5b8845be9970d0Tim Murray llvm::llvm_shutdown_obj Y; 281c46a3f5370364ad5698756d5da5b8845be9970d0Tim Murray init::Initialize(); 282d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao llvm::cl::SetVersionPrinter(BCCVersionPrinter); 283d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao llvm::cl::ParseCommandLineOptions(argc, argv); 284f2ac3176c351cd80bce77fe1488f3de2d0789c1bJean-Luc Brouillet std::string commandLine = bcc::getCommandLine(argc, argv); 285d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 286d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao BCCContext context; 2878be8dba08c97dff239ac8c6bdc7f3a52d96116aeStephen Hines RSCompilerDriver RSCD; 288d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 289c5e607adff80a66bc5420baffd299862abdf368dJean-Luc Brouillet if (OptBCLibFilename.empty()) { 290a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni ALOGE("Failed to compile bitcode, -bclib was not specified"); 291d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao return EXIT_FAILURE; 292d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao } 2938be8dba08c97dff239ac8c6bdc7f3a52d96116aeStephen Hines 2947e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hines if (!ConfigCompiler(RSCD)) { 2957e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hines ALOGE("Failed to configure compiler"); 2967e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hines return EXIT_FAILURE; 2977e9c1853efd1615e9187611e8f47ef3ec30de686Stephen Hines } 298c3437f05c638f8befda59170ae788873db24dc1cStephen Hines 299c3437f05c638f8befda59170ae788873db24dc1cStephen Hines // Attempt to dynamically initialize the compiler driver if such a function 300c3437f05c638f8befda59170ae788873db24dc1cStephen Hines // is present. It is only present if passed via "-load libFOO.so". 301c3437f05c638f8befda59170ae788873db24dc1cStephen Hines RSCompilerDriverInit_t rscdi = (RSCompilerDriverInit_t) 302c3437f05c638f8befda59170ae788873db24dc1cStephen Hines dlsym(RTLD_DEFAULT, STR(RS_COMPILER_DRIVER_INIT_FN)); 303900c6c1f08f7c572125d7d39abe0f0f9eafbfa14Chris Wailes if (rscdi != nullptr) { 304c3437f05c638f8befda59170ae788873db24dc1cStephen Hines rscdi(&RSCD); 305c3437f05c638f8befda59170ae788873db24dc1cStephen Hines } 306c3437f05c638f8befda59170ae788873db24dc1cStephen Hines 3070ab50b835805c75ad164466767c2c212f48954ccYang Ni if (OptMergePlans.size() > 0) { 3080ab50b835805c75ad164466767c2c212f48954ccYang Ni bool success = compileScriptGroup(context, RSCD); 3090ab50b835805c75ad164466767c2c212f48954ccYang Ni 310a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni if (!success) { 311a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni return EXIT_FAILURE; 312a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni } 3130ab50b835805c75ad164466767c2c212f48954ccYang Ni 314a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni return EXIT_SUCCESS; 315a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni } 316a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni 317a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni llvm::ErrorOr<std::unique_ptr<llvm::MemoryBuffer>> mb_or_error = 318a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni llvm::MemoryBuffer::getFile(OptInputFilenames[0].c_str()); 319a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni if (mb_or_error.getError()) { 320a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni ALOGE("Failed to load bitcode from path %s! (%s)", 321a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni OptInputFilenames[0].c_str(), mb_or_error.getError().message().c_str()); 322a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni return EXIT_FAILURE; 323a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni } 324a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni std::unique_ptr<llvm::MemoryBuffer> input_data = std::move(mb_or_error.get()); 325a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni 326a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni const char *bitcode = input_data->getBufferStart(); 327a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni size_t bitcodeSize = input_data->getBufferSize(); 328a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni 3291e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar if (!OptEmbedRSInfo) { 3301e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar bool built = RSCD.build(context, OptOutputPath.c_str(), OptOutputFilename.c_str(), bitcode, 33151ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga Nainar bitcodeSize, commandLine.c_str(), OptChecksum.c_str(), OptBCLibFilename.c_str(), nullptr, 3321e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar OptEmitLLVM); 333d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 3341e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar if (!built) { 3351e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar return EXIT_FAILURE; 3361e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar } 3370ab50b835805c75ad164466767c2c212f48954ccYang Ni } else { 3381e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar // embedRSInfo is set. Use buildForCompatLib to embed RS symbol information 3391e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar // into the .rs.info symbol. 340a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni Source *source = Source::CreateFromBuffer(context, OptInputFilenames[0].c_str(), 3411e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar bitcode, bitcodeSize); 3421e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar RSScript *s = new (std::nothrow) RSScript(*source); 3431e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar if (s == nullptr) { 3441e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar llvm::errs() << "Out of memory when creating script for file `" 345a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni << OptInputFilenames[0] << "'!\n"; 3461e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar delete source; 3471e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar return EXIT_FAILURE; 3481e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar } 3491e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar 3501e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar llvm::SmallString<80> output(OptOutputPath); 3511e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar llvm::sys::path::append(output, "/", OptOutputFilename); 3521e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar llvm::sys::path::replace_extension(output, ".o"); 3531e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar 35451ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga Nainar if (!RSCD.buildForCompatLib(*s, output.c_str(), OptChecksum.c_str(), 35551ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga Nainar OptBCLibFilename.c_str(), OptEmitLLVM)) { 3561e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar fprintf(stderr, "Failed to compile script!"); 3571e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar return EXIT_FAILURE; 3581e0557ae75ae780352845bd2ba0f4babdc5ae4e6Pirama Arumuga Nainar } 359d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao } 360d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao 361d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao return EXIT_SUCCESS; 362d577d11347cdad9b88c0802bf6f5ee1fd3c06f19Shih-wei Liao} 363