Kaushik Kulkarni authored 2 years ago and Andreas Klöckner committed 2 years ago

Co-authored-by: Andreas Kloeckner <inform@tiker.net>

Andreas Klöckner authored 3 years ago and Andreas Klöckner committed 3 years ago

Tweak docstring

Co-authored-by: Alex Fikl <alexfikl@gmail.com>

Kaushik Kulkarni authored 3 years ago and Andreas Klöckner committed 3 years ago

Co-authored-by: Andreas Klöckner <inform@tiker.net>

Closes https://github.com/inducer/arraycontext/issues/70.

* add a leaf_class parameter to flatten

* add some docs to flatten for leaf_class

* allow specifying leaf class in recursive map and map-reduce

* revert broken changes to decorators

* add leaf_class to decorators

* add tests for [multi]mapped_over_array_containers

Co-authored-by: Andreas Klöckner <inform@tiker.net>

Kaushik Kulkarni authored 3 years ago and Andreas Klöckner committed 3 years ago

`compile` only supported compiling callables that returned array
containers. Extends the logic to support compiling callables that simply
return thawed arrays.

Refactor PytatoActx.compile for usability by distributed