roland/xo-alloc
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

xo-jit: + llvm module instrumentation methods

Browse source
This commit is contained in:
Roland Conybeare 2024-06-15 15:12:52 -04:00
commit 3b5193c28d
3 changed files with 34 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

11
include/xo/jit/Jit.hpp
View file

@ -75,6 +75,17 @@ namespace xo {
static llvm::Expected<std::unique_ptr<Jit>> make_aux();
static xo::ref::rp<Jit> make();
// ----- module access -----
/** target triple = string describing target host for codegen **/
const std::string & target_triple() const;
/** append function names defined in attached module to *p_v **/
std::vector<std::string> get_function_name_v();
/** write state of execution session (all the associated dynamic libraries) **/
void dump_execution_session();
// ----- jit code generation -----
llvm::Value * codegen_constant(ref::brw<xo::ast::ConstantInterface> expr);
llvm::Function * codegen_primitive(ref::brw<xo::ast::PrimitiveInterface> expr);

6
include/xo/jit/KaleidoscopeJit.hpp
View file

@ -50,6 +50,7 @@ namespace xo {
using SelfExecutorProcessControl = llvm::orc::SelfExecutorProcessControl;
private:
/* execution session - represents a currenlty-running jit program */
std::unique_ptr<ExecutionSession> ES;
DataLayout DL;
@ -118,6 +119,11 @@ namespace xo {
llvm::Expected<ExecutorSymbolDef> lookup(StringRef Name) {
return ES->lookup({&MainJD}, Mangle(Name.str()));
}
/* dump */
void dump_execution_session() {
ES->dump(llvm::errs());
}
};
} // end namespace jit

17
src/jit/Jit.cpp
View file

@ -127,6 +127,23 @@ namespace xo {
jit_object_layer_.setAutoClaimResponsibilityForObjectSymbols(true);
}
#endif
const std::string &
Jit::target_triple() const {
return llvm_module_->getTargetTriple();
}
std::vector<std::string>
Jit::get_function_name_v() {
std::vector<std::string> retval;
for (const auto & fn_name : *llvm_module_)
retval.push_back(fn_name.getName().str());
return retval;
} /*get_function_names*/
void
Jit::dump_execution_session() {
kal_jit_->dump_execution_session();
}
llvm::Value *