1 //===- IRTransformLayer.h - Run all IR through a functor --------*- C++ -*-===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 // 9 // Run all IR passed in through a user supplied functor. 10 // 11 //===----------------------------------------------------------------------===// 12 13 #ifndef LLVM_EXECUTIONENGINE_ORC_IRTRANSFORMLAYER_H 14 #define LLVM_EXECUTIONENGINE_ORC_IRTRANSFORMLAYER_H 15 16 #include "llvm/ADT/FunctionExtras.h" 17 #include "llvm/ExecutionEngine/JITSymbol.h" 18 #include "llvm/ExecutionEngine/Orc/Layer.h" 19 #include <memory> 20 21 namespace llvm { 22 namespace orc { 23 24 /// A layer that applies a transform to emitted modules. 25 /// The transform function is responsible for locking the ThreadSafeContext 26 /// before operating on the module. 27 class IRTransformLayer : public IRLayer { 28 public: 29 using TransformFunction = unique_function<Expected<ThreadSafeModule>( 30 ThreadSafeModule, MaterializationResponsibility &R)>; 31 32 IRTransformLayer(ExecutionSession &ES, IRLayer &BaseLayer, 33 TransformFunction Transform = identityTransform); 34 35 void setTransform(TransformFunction Transform) { 36 this->Transform = std::move(Transform); 37 } 38 39 void emit(std::unique_ptr<MaterializationResponsibility> R, 40 ThreadSafeModule TSM) override; 41 42 static ThreadSafeModule identityTransform(ThreadSafeModule TSM, 43 MaterializationResponsibility &R) { 44 return TSM; 45 } 46 47 private: 48 IRLayer &BaseLayer; 49 TransformFunction Transform; 50 }; 51 52 } // end namespace orc 53 } // end namespace llvm 54 55 #endif // LLVM_EXECUTIONENGINE_ORC_IRTRANSFORMLAYER_H 56