10fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang/* 20fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang * Copyright 2012, The Android Open Source Project 30fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang * 40fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang * Licensed under the Apache License, Version 2.0 (the "License"); 50fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang * you may not use this file except in compliance with the License. 60fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang * You may obtain a copy of the License at 70fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang * 80fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang * http://www.apache.org/licenses/LICENSE-2.0 90fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang * 100fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang * Unless required by applicable law or agreed to in writing, software 110fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang * distributed under the License is distributed on an "AS IS" BASIS, 120fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 130fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang * See the License for the specific language governing permissions and 140fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang * limitations under the License. 150fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang */ 160fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang 17e198abec6c5e3eab380ccf6897b0a0b9c2dd92ddStephen Hines#include "bcc/Renderscript/RSCompilerDriver.h" 18c72c4ddfcd79c74f70713da91a69569451b5c19eZonr Chang 19a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni#include "llvm/IR/AssemblyAnnotationWriter.h" 207b980e1717f3cf418f7bc4e40597004bc1139b8bTobias Grosser#include <llvm/IR/Module.h> 210ab50b835805c75ad164466767c2c212f48954ccYang Ni#include "llvm/Linker/Linker.h" 22ad69476ec040ba9ee07176ccb758aabb5ff6d51cStephen Hines#include <llvm/Support/CommandLine.h> 23b10c3a7d91556ef31ce91ef018fee4722b783960Stephen Hines#include <llvm/Support/Path.h> 247b980e1717f3cf418f7bc4e40597004bc1139b8bTobias Grosser#include <llvm/Support/raw_ostream.h> 257bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao 267bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao#include "bcinfo/BitcodeWrapper.h" 270ab50b835805c75ad164466767c2c212f48954ccYang Ni#include "bcc/Assert.h" 28a65fba6fd15b14d930809e64c84fb976a893d038Pirama Arumuga Nainar#include "bcinfo/MetadataExtractor.h" 29a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni#include "bcc/BCCContext.h" 3047f0d5a8d9ed629a6d58d3e69891f2c7fdee44d9Stephen Hines#include "bcc/Compiler.h" 31bde1a2599780d6eaebbc284976ceb690492f6abdStephen Hines#include "bcc/Config/Config.h" 32e198abec6c5e3eab380ccf6897b0a0b9c2dd92ddStephen Hines#include "bcc/Renderscript/RSScript.h" 33a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni#include "bcc/Renderscript/RSScriptGroupFusion.h" 34c72c4ddfcd79c74f70713da91a69569451b5c19eZonr Chang#include "bcc/Support/CompilerConfig.h" 357bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao#include "bcc/Source.h" 36c72c4ddfcd79c74f70713da91a69569451b5c19eZonr Chang#include "bcc/Support/FileMutex.h" 37ef73a242762bcd8113b9b65ceccbe7d909b5acbcZonr Chang#include "bcc/Support/Log.h" 38c72c4ddfcd79c74f70713da91a69569451b5c19eZonr Chang#include "bcc/Support/InputFile.h" 39c72c4ddfcd79c74f70713da91a69569451b5c19eZonr Chang#include "bcc/Support/Initialization.h" 40c72c4ddfcd79c74f70713da91a69569451b5c19eZonr Chang#include "bcc/Support/OutputFile.h" 410fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang 420ab50b835805c75ad164466767c2c212f48954ccYang Ni#include <sstream> 43a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni#include <string> 44a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni 45b81d697ec617be24494b622a3532f1c465264415Nick Kralevich#ifdef HAVE_ANDROID_OS 460fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang#include <cutils/properties.h> 47b81d697ec617be24494b622a3532f1c465264415Nick Kralevich#endif 487bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao#include <utils/StopWatch.h> 490fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang 500fffa7e4adc90079dc0f252359050fe872b360ddZonr Changusing namespace bcc; 510fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang 523ab9da1e1b3d7794994393af086d096397fb84dbStephen HinesRSCompilerDriver::RSCompilerDriver(bool pUseCompilerRT) : 53900c6c1f08f7c572125d7d39abe0f0f9eafbfa14Chris Wailes mConfig(nullptr), mCompiler(), mDebugContext(false), 54e67dab471c82c8df1cf73b2fe464792c78082616Stephen Hines mLinkRuntimeCallback(nullptr), mEnableGlobalMerge(true), 55e67dab471c82c8df1cf73b2fe464792c78082616Stephen Hines mEmbedGlobalInfo(false), mEmbedGlobalInfoSkipConstant(false) { 560fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang init::Initialize(); 570fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang} 580fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang 590fffa7e4adc90079dc0f252359050fe872b360ddZonr ChangRSCompilerDriver::~RSCompilerDriver() { 600fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang delete mConfig; 610fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang} 620fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang 630fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang 64bde1a2599780d6eaebbc284976ceb690492f6abdStephen Hines#if defined(PROVIDE_ARM_CODEGEN) 65ad69476ec040ba9ee07176ccb758aabb5ff6d51cStephen Hinesextern llvm::cl::opt<bool> EnableGlobalMerge; 66c06cd06078a8c866d5cecbf897326092b55a8955Stephen Hines#endif 67c06cd06078a8c866d5cecbf897326092b55a8955Stephen Hines 680fffa7e4adc90079dc0f252359050fe872b360ddZonr Changbool RSCompilerDriver::setupConfig(const RSScript &pScript) { 690fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang bool changed = false; 700fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang 710fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang const llvm::CodeGenOpt::Level script_opt_level = 720fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang static_cast<llvm::CodeGenOpt::Level>(pScript.getOptimizationLevel()); 730fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang 74bde1a2599780d6eaebbc284976ceb690492f6abdStephen Hines#if defined(PROVIDE_ARM_CODEGEN) 75045558bd401d0af58a0b57594d75d45ff3035dc9Stephen Hines EnableGlobalMerge = mEnableGlobalMerge; 76045558bd401d0af58a0b57594d75d45ff3035dc9Stephen Hines#endif 77045558bd401d0af58a0b57594d75d45ff3035dc9Stephen Hines 78900c6c1f08f7c572125d7d39abe0f0f9eafbfa14Chris Wailes if (mConfig != nullptr) { 790fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang // Renderscript bitcode may have their optimization flag configuration 800fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang // different than the previous run of RS compilation. 810fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang if (mConfig->getOptimizationLevel() != script_opt_level) { 820fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang mConfig->setOptimizationLevel(script_opt_level); 830fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang changed = true; 840fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang } 850fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang } else { 860fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang // Haven't run the compiler ever. 87bde1a2599780d6eaebbc284976ceb690492f6abdStephen Hines mConfig = new (std::nothrow) CompilerConfig(DEFAULT_TARGET_TRIPLE_STRING); 88900c6c1f08f7c572125d7d39abe0f0f9eafbfa14Chris Wailes if (mConfig == nullptr) { 890fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang // Return false since mConfig remains NULL and out-of-memory. 900fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang return false; 910fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang } 920fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang mConfig->setOptimizationLevel(script_opt_level); 930fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang changed = true; 940fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang } 950fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang 96bde1a2599780d6eaebbc284976ceb690492f6abdStephen Hines#if defined(PROVIDE_ARM_CODEGEN) 97a65fba6fd15b14d930809e64c84fb976a893d038Pirama Arumuga Nainar bcinfo::MetadataExtractor me(&pScript.getSource().getModule()); 98a65fba6fd15b14d930809e64c84fb976a893d038Pirama Arumuga Nainar if (!me.extract()) { 99a65fba6fd15b14d930809e64c84fb976a893d038Pirama Arumuga Nainar assert("Could not extract RS pragma metadata for module!"); 100a65fba6fd15b14d930809e64c84fb976a893d038Pirama Arumuga Nainar } 101a65fba6fd15b14d930809e64c84fb976a893d038Pirama Arumuga Nainar 102a65fba6fd15b14d930809e64c84fb976a893d038Pirama Arumuga Nainar bool script_full_prec = (me.getRSFloatPrecision() == bcinfo::RS_FP_Full); 103bde1a2599780d6eaebbc284976ceb690492f6abdStephen Hines if (mConfig->getFullPrecision() != script_full_prec) { 104bde1a2599780d6eaebbc284976ceb690492f6abdStephen Hines mConfig->setFullPrecision(script_full_prec); 105bde1a2599780d6eaebbc284976ceb690492f6abdStephen Hines changed = true; 1060fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang } 1070fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang#endif 1080fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang 1090fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang return changed; 1100fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang} 1110fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang 112f2ac3176c351cd80bce77fe1488f3de2d0789c1bJean-Luc BrouilletCompiler::ErrorCode RSCompilerDriver::compileScript(RSScript& pScript, const char* pScriptName, 113f2ac3176c351cd80bce77fe1488f3de2d0789c1bJean-Luc Brouillet const char* pOutputPath, 114f2ac3176c351cd80bce77fe1488f3de2d0789c1bJean-Luc Brouillet const char* pRuntimePath, 11551ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga Nainar const char* pBuildChecksum, 116a65fba6fd15b14d930809e64c84fb976a893d038Pirama Arumuga Nainar bool pDumpIR) { 11751ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga Nainar // embed build checksum metadata into the source 11851ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga Nainar if (pBuildChecksum != nullptr && strlen(pBuildChecksum) > 0) { 11951ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga Nainar pScript.getSource().addBuildChecksumMetadata(pBuildChecksum); 12051ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga Nainar } 12151ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga Nainar 122cb6c159729e7a9f2582e2758ff45e75bac5a00c0Pirama Arumuga Nainar // Verify that the only external functions in pScript are Renderscript 123cb6c159729e7a9f2582e2758ff45e75bac5a00c0Pirama Arumuga Nainar // functions. Fail if verification returns an error. 124cb6c159729e7a9f2582e2758ff45e75bac5a00c0Pirama Arumuga Nainar if (mCompiler.screenGlobalFunctions(pScript) != Compiler::kSuccess) { 125cb6c159729e7a9f2582e2758ff45e75bac5a00c0Pirama Arumuga Nainar return Compiler::kErrInvalidSource; 126cb6c159729e7a9f2582e2758ff45e75bac5a00c0Pirama Arumuga Nainar } 127cb6c159729e7a9f2582e2758ff45e75bac5a00c0Pirama Arumuga Nainar 1280fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang //===--------------------------------------------------------------------===// 129e198abec6c5e3eab380ccf6897b0a0b9c2dd92ddStephen Hines // Link RS script with Renderscript runtime. 130ba4206445e192f4c0fa519eae55c4b84cfb0efeeShih-wei Liao //===--------------------------------------------------------------------===// 131331310e1f3f86a795f78e42b3f03558a43829f09Stephen Hines if (!RSScript::LinkRuntime(pScript, pRuntimePath)) { 1326da4e253a513feef3405759fef6d0760828808caYang Ni ALOGE("Failed to link script '%s' with Renderscript runtime %s!", 1336da4e253a513feef3405759fef6d0760828808caYang Ni pScriptName, pRuntimePath); 13447f0d5a8d9ed629a6d58d3e69891f2c7fdee44d9Stephen Hines return Compiler::kErrInvalidSource; 135ba4206445e192f4c0fa519eae55c4b84cfb0efeeShih-wei Liao } 136ba4206445e192f4c0fa519eae55c4b84cfb0efeeShih-wei Liao 13701f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines { 1380784365a38fd5a9c08dd484f9ba549328d4dff97Stephen Hines // FIXME(srhines): Windows compilation can't use locking like this, but 1390784365a38fd5a9c08dd484f9ba549328d4dff97Stephen Hines // we also don't need to worry about concurrent writers of the same file. 140d7a95265621a788a6de8c0c3a27a588d7c466f79Stephen Hines#ifndef USE_MINGW 1410784365a38fd5a9c08dd484f9ba549328d4dff97Stephen Hines //===------------------------------------------------------------------===// 14201f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines // Acquire the write lock for writing output object file. 1430784365a38fd5a9c08dd484f9ba549328d4dff97Stephen Hines //===------------------------------------------------------------------===// 14401f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines FileMutex<FileBase::kWriteLock> write_output_mutex(pOutputPath); 1450fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang 14601f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines if (write_output_mutex.hasError() || !write_output_mutex.lock()) { 14701f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines ALOGE("Unable to acquire the lock for writing %s! (%s)", 14801f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines pOutputPath, write_output_mutex.getErrorMessage().c_str()); 14947f0d5a8d9ed629a6d58d3e69891f2c7fdee44d9Stephen Hines return Compiler::kErrInvalidSource; 15001f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines } 151d7a95265621a788a6de8c0c3a27a588d7c466f79Stephen Hines#endif 1520fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang 15301f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines // Open the output file for write. 154acf9c9eeb5a07535dbed5b272c7f22cc1b050a40Stephen Hines OutputFile output_file(pOutputPath, 155acf9c9eeb5a07535dbed5b272c7f22cc1b050a40Stephen Hines FileBase::kTruncate | FileBase::kBinary); 1560fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang 15701f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines if (output_file.hasError()) { 15801f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines ALOGE("Unable to open %s for write! (%s)", pOutputPath, 15901f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines output_file.getErrorMessage().c_str()); 16047f0d5a8d9ed629a6d58d3e69891f2c7fdee44d9Stephen Hines return Compiler::kErrInvalidSource; 16101f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines } 1620fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang 16301f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines // Setup the config to the compiler. 16401f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines bool compiler_need_reconfigure = setupConfig(pScript); 1650fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang 166900c6c1f08f7c572125d7d39abe0f0f9eafbfa14Chris Wailes if (mConfig == nullptr) { 16701f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines ALOGE("Failed to setup config for RS compiler to compile %s!", 16801f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines pOutputPath); 16947f0d5a8d9ed629a6d58d3e69891f2c7fdee44d9Stephen Hines return Compiler::kErrInvalidSource; 1700fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang } 1710fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang 17201f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines if (compiler_need_reconfigure) { 17301f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines Compiler::ErrorCode err = mCompiler.config(*mConfig); 17401f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines if (err != Compiler::kSuccess) { 17501f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines ALOGE("Failed to config the RS compiler for %s! (%s)",pOutputPath, 17601f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines Compiler::GetErrorString(err)); 17747f0d5a8d9ed629a6d58d3e69891f2c7fdee44d9Stephen Hines return Compiler::kErrInvalidSource; 17801f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines } 17901f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines } 18001f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines 181900c6c1f08f7c572125d7d39abe0f0f9eafbfa14Chris Wailes OutputFile *ir_file = nullptr; 182900c6c1f08f7c572125d7d39abe0f0f9eafbfa14Chris Wailes llvm::raw_fd_ostream *IRStream = nullptr; 18327fb7edfd3f53f52fba6ee81267c02f7896198a3Tobias Grosser if (pDumpIR) { 18435978e7a7e87b7147789ff3ca9c3a6bf7dd165b0Chris Wailes std::string path(pOutputPath); 18527fb7edfd3f53f52fba6ee81267c02f7896198a3Tobias Grosser path.append(".ll"); 18635978e7a7e87b7147789ff3ca9c3a6bf7dd165b0Chris Wailes ir_file = new OutputFile(path.c_str(), FileBase::kTruncate); 18727fb7edfd3f53f52fba6ee81267c02f7896198a3Tobias Grosser IRStream = ir_file->dup(); 18827fb7edfd3f53f52fba6ee81267c02f7896198a3Tobias Grosser } 18927fb7edfd3f53f52fba6ee81267c02f7896198a3Tobias Grosser 19001f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines // Run the compiler. 19135978e7a7e87b7147789ff3ca9c3a6bf7dd165b0Chris Wailes Compiler::ErrorCode compile_result = 19235978e7a7e87b7147789ff3ca9c3a6bf7dd165b0Chris Wailes mCompiler.compile(pScript, output_file, IRStream); 19327fb7edfd3f53f52fba6ee81267c02f7896198a3Tobias Grosser 19427fb7edfd3f53f52fba6ee81267c02f7896198a3Tobias Grosser if (ir_file) { 19527fb7edfd3f53f52fba6ee81267c02f7896198a3Tobias Grosser ir_file->close(); 19627fb7edfd3f53f52fba6ee81267c02f7896198a3Tobias Grosser delete ir_file; 19727fb7edfd3f53f52fba6ee81267c02f7896198a3Tobias Grosser } 19801f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines 19901f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines if (compile_result != Compiler::kSuccess) { 20001f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines ALOGE("Unable to compile the source to file %s! (%s)", pOutputPath, 20101f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines Compiler::GetErrorString(compile_result)); 20247f0d5a8d9ed629a6d58d3e69891f2c7fdee44d9Stephen Hines return Compiler::kErrInvalidSource; 20301f05d4b45cbde1e07d4707152908c1d843f1328Stephen Hines } 2040fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang } 2050fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang 20647f0d5a8d9ed629a6d58d3e69891f2c7fdee44d9Stephen Hines return Compiler::kSuccess; 2070fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang} 2080fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang 20947f0d5a8d9ed629a6d58d3e69891f2c7fdee44d9Stephen Hinesbool RSCompilerDriver::build(BCCContext &pContext, 21047f0d5a8d9ed629a6d58d3e69891f2c7fdee44d9Stephen Hines const char *pCacheDir, 21147f0d5a8d9ed629a6d58d3e69891f2c7fdee44d9Stephen Hines const char *pResName, 21247f0d5a8d9ed629a6d58d3e69891f2c7fdee44d9Stephen Hines const char *pBitcode, 21347f0d5a8d9ed629a6d58d3e69891f2c7fdee44d9Stephen Hines size_t pBitcodeSize, 21451ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga Nainar const char *pBuildChecksum, 21547f0d5a8d9ed629a6d58d3e69891f2c7fdee44d9Stephen Hines const char *pRuntimePath, 2167b980e1717f3cf418f7bc4e40597004bc1139b8bTobias Grosser RSLinkRuntimeCallback pLinkRuntimeCallback, 2177b980e1717f3cf418f7bc4e40597004bc1139b8bTobias Grosser bool pDumpIR) { 218c89f78b40e7823eb2d08e7129794c9f6f1992c08Tim Murray // android::StopWatch build_time("bcc: RSCompilerDriver::build time"); 2197bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao //===--------------------------------------------------------------------===// 2207bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao // Check parameters. 2217bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao //===--------------------------------------------------------------------===// 222900c6c1f08f7c572125d7d39abe0f0f9eafbfa14Chris Wailes if ((pCacheDir == nullptr) || (pResName == nullptr)) { 2237bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao ALOGE("Invalid parameter passed to RSCompilerDriver::build()! (cache dir: " 2247bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao "%s, resource name: %s)", ((pCacheDir) ? pCacheDir : "(null)"), 2257bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao ((pResName) ? pResName : "(null)")); 22647f0d5a8d9ed629a6d58d3e69891f2c7fdee44d9Stephen Hines return false; 2277bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao } 2287bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao 229900c6c1f08f7c572125d7d39abe0f0f9eafbfa14Chris Wailes if ((pBitcode == nullptr) || (pBitcodeSize <= 0)) { 2307bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao ALOGE("No bitcode supplied! (bitcode: %p, size of bitcode: %u)", 2317bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao pBitcode, static_cast<unsigned>(pBitcodeSize)); 23247f0d5a8d9ed629a6d58d3e69891f2c7fdee44d9Stephen Hines return false; 2337bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao } 2347bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao 2357bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao //===--------------------------------------------------------------------===// 2367bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao // Construct output path. 2377bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao // {pCacheDir}/{pResName}.o 238b10c3a7d91556ef31ce91ef018fee4722b783960Stephen Hines //===--------------------------------------------------------------------===// 239b10c3a7d91556ef31ce91ef018fee4722b783960Stephen Hines llvm::SmallString<80> output_path(pCacheDir); 240b10c3a7d91556ef31ce91ef018fee4722b783960Stephen Hines llvm::sys::path::append(output_path, pResName); 241b10c3a7d91556ef31ce91ef018fee4722b783960Stephen Hines llvm::sys::path::replace_extension(output_path, ".o"); 2427bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao 2437bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao //===--------------------------------------------------------------------===// 2447bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao // Load the bitcode and create script. 2457bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao //===--------------------------------------------------------------------===// 2467bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao Source *source = Source::CreateFromBuffer(pContext, pResName, 2477bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao pBitcode, pBitcodeSize); 248900c6c1f08f7c572125d7d39abe0f0f9eafbfa14Chris Wailes if (source == nullptr) { 24947f0d5a8d9ed629a6d58d3e69891f2c7fdee44d9Stephen Hines return false; 2507bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao } 2517bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao 252f2ac3176c351cd80bce77fe1488f3de2d0789c1bJean-Luc Brouillet RSScript script(*source); 253c3437f05c638f8befda59170ae788873db24dc1cStephen Hines if (pLinkRuntimeCallback) { 254c3437f05c638f8befda59170ae788873db24dc1cStephen Hines setLinkRuntimeCallback(pLinkRuntimeCallback); 255c3437f05c638f8befda59170ae788873db24dc1cStephen Hines } 2567bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao 257f2ac3176c351cd80bce77fe1488f3de2d0789c1bJean-Luc Brouillet script.setLinkRuntimeCallback(getLinkRuntimeCallback()); 25806731a6150ae8014d37258d5f32ef8bc14a3db63Stephen Hines 259e67dab471c82c8df1cf73b2fe464792c78082616Stephen Hines script.setEmbedGlobalInfo(mEmbedGlobalInfo); 260e67dab471c82c8df1cf73b2fe464792c78082616Stephen Hines script.setEmbedGlobalInfoSkipConstant(mEmbedGlobalInfoSkipConstant); 261e67dab471c82c8df1cf73b2fe464792c78082616Stephen Hines 2627bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao // Read information from bitcode wrapper. 2637bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao bcinfo::BitcodeWrapper wrapper(pBitcode, pBitcodeSize); 264f2ac3176c351cd80bce77fe1488f3de2d0789c1bJean-Luc Brouillet script.setCompilerVersion(wrapper.getCompilerVersion()); 265f2ac3176c351cd80bce77fe1488f3de2d0789c1bJean-Luc Brouillet script.setOptimizationLevel(static_cast<RSScript::OptimizationLevel>( 266f2ac3176c351cd80bce77fe1488f3de2d0789c1bJean-Luc Brouillet wrapper.getOptimizationLevel())); 2677bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao 2687bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao //===--------------------------------------------------------------------===// 2697bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao // Compile the script 2707bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao //===--------------------------------------------------------------------===// 271f2ac3176c351cd80bce77fe1488f3de2d0789c1bJean-Luc Brouillet Compiler::ErrorCode status = compileScript(script, pResName, 27247f0d5a8d9ed629a6d58d3e69891f2c7fdee44d9Stephen Hines output_path.c_str(), 273a65fba6fd15b14d930809e64c84fb976a893d038Pirama Arumuga Nainar pRuntimePath, 274a65fba6fd15b14d930809e64c84fb976a893d038Pirama Arumuga Nainar pBuildChecksum, 275a65fba6fd15b14d930809e64c84fb976a893d038Pirama Arumuga Nainar pDumpIR); 2767bcec85228c1eceade6efc9cd7d02ef81c8d55e9Shih-wei Liao 277f2ac3176c351cd80bce77fe1488f3de2d0789c1bJean-Luc Brouillet return status == Compiler::kSuccess; 2780fffa7e4adc90079dc0f252359050fe872b360ddZonr Chang} 279331310e1f3f86a795f78e42b3f03558a43829f09Stephen Hines 280a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Nibool RSCompilerDriver::buildScriptGroup( 2810ab50b835805c75ad164466767c2c212f48954ccYang Ni BCCContext& Context, const char* pOutputFilepath, const char* pRuntimePath, 282186d2f34573eb20de197d8eb8e08bcadd031a9bdYang Ni const char* pRuntimeRelaxedPath, bool dumpIR, const char* buildChecksum, 2836da4e253a513feef3405759fef6d0760828808caYang Ni const std::vector<Source*>& sources, 2840ab50b835805c75ad164466767c2c212f48954ccYang Ni const std::list<std::list<std::pair<int, int>>>& toFuse, 2850ab50b835805c75ad164466767c2c212f48954ccYang Ni const std::list<std::string>& fused, 2860ab50b835805c75ad164466767c2c212f48954ccYang Ni const std::list<std::list<std::pair<int, int>>>& invokes, 2870ab50b835805c75ad164466767c2c212f48954ccYang Ni const std::list<std::string>& invokeBatchNames) { 2880ab50b835805c75ad164466767c2c212f48954ccYang Ni // --------------------------------------------------------------------------- 2890ab50b835805c75ad164466767c2c212f48954ccYang Ni // Link all input modules into a single module 2900ab50b835805c75ad164466767c2c212f48954ccYang Ni // --------------------------------------------------------------------------- 2910ab50b835805c75ad164466767c2c212f48954ccYang Ni 2920ab50b835805c75ad164466767c2c212f48954ccYang Ni llvm::LLVMContext& context = Context.getLLVMContext(); 2930ab50b835805c75ad164466767c2c212f48954ccYang Ni llvm::Module module("Merged Script Group", context); 2940ab50b835805c75ad164466767c2c212f48954ccYang Ni 2950ab50b835805c75ad164466767c2c212f48954ccYang Ni llvm::Linker linker(&module); 2960ab50b835805c75ad164466767c2c212f48954ccYang Ni for (Source* source : sources) { 2970ab50b835805c75ad164466767c2c212f48954ccYang Ni if (linker.linkInModule(&source->getModule())) { 2980ab50b835805c75ad164466767c2c212f48954ccYang Ni ALOGE("Linking for module in source failed."); 2990ab50b835805c75ad164466767c2c212f48954ccYang Ni return false; 3000ab50b835805c75ad164466767c2c212f48954ccYang Ni } 3010ab50b835805c75ad164466767c2c212f48954ccYang Ni } 3020ab50b835805c75ad164466767c2c212f48954ccYang Ni 3030ab50b835805c75ad164466767c2c212f48954ccYang Ni // --------------------------------------------------------------------------- 3040ab50b835805c75ad164466767c2c212f48954ccYang Ni // Create fused kernels 3050ab50b835805c75ad164466767c2c212f48954ccYang Ni // --------------------------------------------------------------------------- 3060ab50b835805c75ad164466767c2c212f48954ccYang Ni 3070ab50b835805c75ad164466767c2c212f48954ccYang Ni auto inputIter = toFuse.begin(); 3080ab50b835805c75ad164466767c2c212f48954ccYang Ni for (const std::string& nameOfFused : fused) { 3090ab50b835805c75ad164466767c2c212f48954ccYang Ni auto inputKernels = *inputIter++; 3100ab50b835805c75ad164466767c2c212f48954ccYang Ni std::vector<Source*> sourcesToFuse; 3110ab50b835805c75ad164466767c2c212f48954ccYang Ni std::vector<int> slots; 3120ab50b835805c75ad164466767c2c212f48954ccYang Ni 3130ab50b835805c75ad164466767c2c212f48954ccYang Ni for (auto p : inputKernels) { 3140ab50b835805c75ad164466767c2c212f48954ccYang Ni sourcesToFuse.push_back(sources[p.first]); 3150ab50b835805c75ad164466767c2c212f48954ccYang Ni slots.push_back(p.second); 3160ab50b835805c75ad164466767c2c212f48954ccYang Ni } 3170ab50b835805c75ad164466767c2c212f48954ccYang Ni 3180ab50b835805c75ad164466767c2c212f48954ccYang Ni if (!fuseKernels(Context, sourcesToFuse, slots, nameOfFused, &module)) { 3190ab50b835805c75ad164466767c2c212f48954ccYang Ni return false; 3200ab50b835805c75ad164466767c2c212f48954ccYang Ni } 321a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni } 322a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni 3230ab50b835805c75ad164466767c2c212f48954ccYang Ni // --------------------------------------------------------------------------- 3240ab50b835805c75ad164466767c2c212f48954ccYang Ni // Rename invokes 3250ab50b835805c75ad164466767c2c212f48954ccYang Ni // --------------------------------------------------------------------------- 3260ab50b835805c75ad164466767c2c212f48954ccYang Ni 3270ab50b835805c75ad164466767c2c212f48954ccYang Ni auto invokeIter = invokes.begin(); 3280ab50b835805c75ad164466767c2c212f48954ccYang Ni for (const std::string& newName : invokeBatchNames) { 3290ab50b835805c75ad164466767c2c212f48954ccYang Ni auto inputInvoke = *invokeIter++; 3300ab50b835805c75ad164466767c2c212f48954ccYang Ni auto p = inputInvoke.front(); 3310ab50b835805c75ad164466767c2c212f48954ccYang Ni Source* source = sources[p.first]; 3320ab50b835805c75ad164466767c2c212f48954ccYang Ni int slot = p.second; 3330ab50b835805c75ad164466767c2c212f48954ccYang Ni 3340ab50b835805c75ad164466767c2c212f48954ccYang Ni if (!renameInvoke(Context, source, slot, newName, &module)) { 3350ab50b835805c75ad164466767c2c212f48954ccYang Ni return false; 3360ab50b835805c75ad164466767c2c212f48954ccYang Ni } 3370ab50b835805c75ad164466767c2c212f48954ccYang Ni } 3380ab50b835805c75ad164466767c2c212f48954ccYang Ni 3390ab50b835805c75ad164466767c2c212f48954ccYang Ni // --------------------------------------------------------------------------- 3400ab50b835805c75ad164466767c2c212f48954ccYang Ni // Compile the new module with fused kernels 3410ab50b835805c75ad164466767c2c212f48954ccYang Ni // --------------------------------------------------------------------------- 3420ab50b835805c75ad164466767c2c212f48954ccYang Ni 343a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni const std::unique_ptr<Source> source( 3440ab50b835805c75ad164466767c2c212f48954ccYang Ni Source::CreateFromModule(Context, pOutputFilepath, module, true)); 345a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni RSScript script(*source); 346a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni 3470ab50b835805c75ad164466767c2c212f48954ccYang Ni // Embed the info string directly in the ELF 3480ab50b835805c75ad164466767c2c212f48954ccYang Ni script.setEmbedInfo(true); 3490ab50b835805c75ad164466767c2c212f48954ccYang Ni script.setOptimizationLevel(RSScript::kOptLvl3); 350e67dab471c82c8df1cf73b2fe464792c78082616Stephen Hines script.setEmbedGlobalInfo(mEmbedGlobalInfo); 351e67dab471c82c8df1cf73b2fe464792c78082616Stephen Hines script.setEmbedGlobalInfoSkipConstant(mEmbedGlobalInfoSkipConstant); 352a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni 353a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni llvm::SmallString<80> output_path(pOutputFilepath); 354a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni llvm::sys::path::replace_extension(output_path, ".o"); 355a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni 3566da4e253a513feef3405759fef6d0760828808caYang Ni // Pick the right runtime lib 3576da4e253a513feef3405759fef6d0760828808caYang Ni const char* coreLibPath = pRuntimePath; 3586da4e253a513feef3405759fef6d0760828808caYang Ni if (strcmp(pRuntimeRelaxedPath, "")) { 3596da4e253a513feef3405759fef6d0760828808caYang Ni bcinfo::MetadataExtractor me(&module); 3606da4e253a513feef3405759fef6d0760828808caYang Ni me.extract(); 3616da4e253a513feef3405759fef6d0760828808caYang Ni if (me.getRSFloatPrecision() == bcinfo::RS_FP_Relaxed) { 3626da4e253a513feef3405759fef6d0760828808caYang Ni coreLibPath = pRuntimeRelaxedPath; 3636da4e253a513feef3405759fef6d0760828808caYang Ni } 3646da4e253a513feef3405759fef6d0760828808caYang Ni } 3656da4e253a513feef3405759fef6d0760828808caYang Ni 3666da4e253a513feef3405759fef6d0760828808caYang Ni compileScript(script, pOutputFilepath, output_path.c_str(), coreLibPath, 367a65fba6fd15b14d930809e64c84fb976a893d038Pirama Arumuga Nainar buildChecksum, dumpIR); 368a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni 369a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni return true; 370a4ded1373d7ad3e503f186e65bccf94126a0f020Yang Ni} 371331310e1f3f86a795f78e42b3f03558a43829f09Stephen Hines 372c5e607adff80a66bc5420baffd299862abdf368dJean-Luc Brouilletbool RSCompilerDriver::buildForCompatLib(RSScript &pScript, const char *pOut, 37351ee77bd31e7d8ca6c89e37b5806c8fc2afcf0dcPirama Arumuga Nainar const char *pBuildChecksum, 37477c1d553e5799aef5add08667e71b62b6842ed2eTim Murray const char *pRuntimePath, 37577c1d553e5799aef5add08667e71b62b6842ed2eTim Murray bool pDumpIR) { 37686a0b797c221d4c3373dc10c8229b75b6747f6e7Stephen Hines // Embed the info string directly in the ELF, since this path is for an 37786a0b797c221d4c3373dc10c8229b75b6747f6e7Stephen Hines // offline (host) compilation. 37886a0b797c221d4c3373dc10c8229b75b6747f6e7Stephen Hines pScript.setEmbedInfo(true); 37986a0b797c221d4c3373dc10c8229b75b6747f6e7Stephen Hines 380e67dab471c82c8df1cf73b2fe464792c78082616Stephen Hines pScript.setEmbedGlobalInfo(mEmbedGlobalInfo); 381e67dab471c82c8df1cf73b2fe464792c78082616Stephen Hines pScript.setEmbedGlobalInfoSkipConstant(mEmbedGlobalInfoSkipConstant); 382d47777179abaffa360d4ba6a59c19498c61350c1Stephen Hines pScript.setLinkRuntimeCallback(getLinkRuntimeCallback()); 383e67dab471c82c8df1cf73b2fe464792c78082616Stephen Hines 3846da4e253a513feef3405759fef6d0760828808caYang Ni Compiler::ErrorCode status = compileScript(pScript, pOut, pOut, pRuntimePath, 385a65fba6fd15b14d930809e64c84fb976a893d038Pirama Arumuga Nainar pBuildChecksum, pDumpIR); 38647f0d5a8d9ed629a6d58d3e69891f2c7fdee44d9Stephen Hines if (status != Compiler::kSuccess) { 38747f0d5a8d9ed629a6d58d3e69891f2c7fdee44d9Stephen Hines return false; 38847f0d5a8d9ed629a6d58d3e69891f2c7fdee44d9Stephen Hines } 38947f0d5a8d9ed629a6d58d3e69891f2c7fdee44d9Stephen Hines 39047f0d5a8d9ed629a6d58d3e69891f2c7fdee44d9Stephen Hines return true; 391331310e1f3f86a795f78e42b3f03558a43829f09Stephen Hines} 392