search

elixir-nx / nx

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

feat(exla): add LU custom_call #1549

Closed polvalente closed 3 weeks ago

polvalente commented 3 weeks ago

Adds a CPU custom call for Nx.LinAlg.lu, as that function is needed for Nx.LinAlg.determinant

polvalente commented 3 weeks ago

This is related to, but doesn't fully solve, #1388

polvalente commented 3 weeks ago

@josevalim yeah, that solves most of the problem already. Having the defn definition would be a lot better, so that's why I don't want to close the issue.