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

xo-pyjit: + lookup_dbl_dbl_fn [wip -- not actually working yet]

Browse source
This commit is contained in:
Roland Conybeare 2024-06-15 15:16:00 -04:00
commit 8e72f98bcc
1 changed files with 11 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

12
src/pyjit/pyjit.cpp
View file

@ -49,9 +49,19 @@ namespace xo {
/* we're assuming llvm-generated code lives for as long as the Jit
* instance that created it.
*
* RC 14jun2024 - I think this is true modulo use of llvm resource trackers.
* RC 14jun2024 - I think this is true, modulo use of llvm resource trackers.
*/
py::return_value_policy::reference_internal)
.def("lookup_dbl_dbl_fn",
[](Jit & jit, const std::string & symbol) {
auto llvm_addr = jit.lookup_symbol(symbol);
auto fn_addr = llvm_addr.toPtr<double (*) (double)>();
return new XferFn(fn_addr);
})
;
py::class_<XferFn, rp<XferFn>>(m, "XferFn")
.def("__call__",
[](XferFn & self, double x) { return self(x); }