Make Dex functions callable from Haskell

google-research / dex-lang

Research language for array processing in the Haskell/ML family

BSD 3-Clause "New" or "Revised" License

1.56k stars 106 forks source link

Make Dex functions callable from Haskell #1256

Closed axch closed 1 year ago

axch commented 1 year ago

Goal: Be able to unit-test an upcoming experimental jax->dex pipeline from Haskell.

Side-benefit: Speed up our Haskell unit test suite (and improve coverage) by not recompiling the trivial casting function with LLVM for every input value on which we might test its behavior.

This occasioned a rework of TopLevel.hs, to make the actual compilation pipeline clearer and more uniform.