search

elixir-nx / nx

Multi-dimensional arrays (tensors) and numerical definitions for Elixir
2.66k stars 194 forks source link

fix: add support for defn compilation in EXLA.to_mlir_module #1530

Closed polvalente closed 2 months ago

polvalente commented 2 months ago

EXLA.to_mlir_module wasn't working when being called from within a Defn compiler. The new option adds the control needed for this to work.