bffb492c1c
Splits native codegen into a build phase (needs cc) and a deploy phase (none):
- gen-c-module/compile-module compile MANY numeric-leaf fns into ONE native
module (the AOT shape), generalizing the one-fn-per-.so JIT path.
- Backend :cgen-collect? records each numeric-leaf defn's IR while the app loads
as bytecode; cgen/aot-build compiles them into one module and write-manifest
persists {sopath, [{ns name sym}]}.
- Backend :cgen-prebuilt + cgen/load-aot: the deploy run loads the prebuilt .so
(via the native builtin, no cc) and installs each cfunction as the var root
with the same timing as the JIT path, so callers direct-link to native code.
- toolchain-available? no longer crashes when cc is off PATH (os/execute raises
on a missing exe) — a toolchain-less target now gets false.
Proven end-to-end in two processes (spike/native/aot-demo.janet): build with cc,
then deploy with cc removed from PATH -> count-point still native, mandelbrot
3288753 at 12.4ms (full 18x). Test: test/integration/cgen-aot-test.janet. Default
path unchanged; the modes are opt-in. Gate green (118 files).
Remaining for a literal single binary: fuse the .so + manifest into the jpm exe.
Co-authored-by: Yogthos <yogthos@gmail.com>
47 lines
2.2 KiB
Text
47 lines
2.2 KiB
Text
# AOT build/deploy demo (jolt-a7ds). Two phases, run as separate processes:
|
|
# janet spike/native/aot-demo.janet build # needs cc; compiles + writes manifest
|
|
# janet spike/native/aot-demo.janet deploy # run with cc REMOVED from PATH
|
|
# Proves the deploy target needs no C toolchain: it only loads the prebuilt .so.
|
|
(import ../../src/jolt/api :as api)
|
|
(import ../../src/jolt/cgen :as cgen)
|
|
|
|
(def cp "(defn count-point [cr ci cap] (loop [i 0 zr 0.0 zi 0.0] (if (or (>= i cap) (> (+ (* zr zr) (* zi zi)) 4.0)) i (recur (inc i) (+ (- (* zr zr) (* zi zi)) cr) (+ (* 2.0 (* zr zi)) ci)))))")
|
|
(def run "(defn run [n] (let [cap 200 nd (* 1.0 n)] (loop [y 0 acc 0] (if (< y n) (let [ci (- (/ (* 2.0 y) nd) 1.0) row (loop [x 0 a 0] (if (< x n) (let [cr (- (/ (* 2.0 x) nd) 1.5)] (recur (inc x) (+ a (count-point cr ci cap)))) a))] (recur (inc y) (+ acc row))) acc))))")
|
|
(def manifest (string (or (os/getenv "TMPDIR") "/tmp") "/jolt-aot-demo.jdn"))
|
|
(def dir "spike/native/build/aot")
|
|
|
|
(defn setup [ctx]
|
|
(put (ctx :env) :direct-linking? true)
|
|
(api/eval-string ctx "(ns demo)"))
|
|
|
|
(def phase (get (dyn :args) 1))
|
|
(cond
|
|
(= phase "build")
|
|
(do
|
|
(def ctx (api/init-cached {:compile? true}))
|
|
(put (ctx :env) :cgen-collect? true)
|
|
(setup ctx)
|
|
(api/eval-string ctx cp)
|
|
(api/eval-string ctx run)
|
|
(def build (cgen/aot-build (get (ctx :env) :cgen-collected) {:dir dir}))
|
|
(cgen/write-manifest manifest build)
|
|
(printf "BUILD: cc-available? %p -> %s (%d fn)" (cgen/toolchain-available?)
|
|
(build :sopath) (length (build :entries))))
|
|
|
|
(= phase "deploy")
|
|
(do
|
|
(printf "DEPLOY: cc-available? %p (should be false with cc off PATH)"
|
|
(cgen/toolchain-available?))
|
|
(def ctx (api/init-cached {:compile? true}))
|
|
(setup ctx)
|
|
(put (ctx :env) :cgen-prebuilt (cgen/load-aot manifest))
|
|
(api/eval-string ctx cp)
|
|
(api/eval-string ctx run)
|
|
(def native? (cfunction? (api/eval-string ctx "count-point")))
|
|
(def t0 (os/clock))
|
|
(def total (api/eval-string ctx "(run 200)"))
|
|
(def ms (* 1000 (- (os/clock) t0)))
|
|
(printf "DEPLOY: count-point native? %p total %d (%s) %.1f ms"
|
|
native? total (if (= total 3288753) "OK" "MISMATCH") ms))
|
|
|
|
(eprint "usage: aot-demo.janet build|deploy"))
|