* LazilyPyOpenCLCompilingFunctionCaller: limit arg size for GPUs

* move limit

* also check for SVM presence

* get_target()

* memoize get_target

* UNDO BEFORE MERGE: use dev branches

* Hackety hack: SVM detection in actx constructor

* check whether passed allocator supports SVM

* undo loopy branch

* implement it for the base class

* subclass LoopyPyOpenCLTarget

* set actual limit

* undo pytato branch

* remove unused argument

* add type annotations

* add logging

* Refactor arg size passing to put less logic in the target

* flake8

* add a test

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

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)

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

* adds an is_available to check is the array context works
* do not create a factory per CL device, since that does not
quite work for non-CL array contexts

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>