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>

* broadcast arithmetic if arg2 is actx's array

* define and implement get_array_types

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

* adds test_leaf_array_type_broadcasting

* avoid using iff in the docs

replaced with only if

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

* adds docs for test, tests with scalars

* ArrayContext.get_array_types() -> ArrayContext.array_types

* bcast_actx_array_type -> _bcast_actx_array_type

Co-authored-by: Andreas Kloeckner <andreask@illinois.edu>

* formatting: remove unused import

* leaf array bcast types: better code placement

* make array_types a property

* docs: grammar

* better var naming: ary_types -> bcast_actx_ary_types

* Revert "make array_types a property"

This reverts commit e6b8b1b1860aadd6cd95269ca18dc22303b5b8b5.

* ArrayContext: make array_types a class attribute

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