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

Co-authored-by: Michael Campbell <mtcampbe@illinois.edu>

* support empty containers in array_equal

* support empty containers in array_equal (jax)

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

Co-authored-by: Kaushik Kulkarni <kaushikcfd@gmail.com>

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

Needs https://github.com/inducer/loopy/pull/624.

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>

* 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>

* Make broadcast settings visible on users of with_container_arithmetic

* Remove stray debug print

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

* Fix _outer_bcast_types attribute name

* add alternate outer product

* add empty line

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

* add comment explaining _outer_bcast_types

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

* add more detail to docstring first line

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

* rename is_scalar -> treat_as_scalar

* treat non-object numpy arrays as scalars in outer

* remove use of deprecated is_array_container

* clarify usage of isinstance(..., ndarray) for object array detection

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

Co-authored-by: Andreas Kloeckner <inform@tiker.net>
Co-authored-by: Alex Fikl <alexfikl@gmail.com>

* add recursive container reductions

* extend reductions to array containers

* make rec_reduce_array_container more specific

* rename make_container to process_container

* rename rec_reduce_container to rec_map_reduce_container

* fix typo in comment

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

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