4935713c7fad540e1c461297e9e9de60194533c4 to main · Andreas Klöckner / arraycontext · GitLab

Commits on Source 356

33884ae5 expand tests to actually check scalars
Nov 07, 2021
36fd05c6 Refactor PytatoActx.compile for usability by distributed
Nov 17, 2021
35991c2a Merge pull request #124 from inducer/refactor-pytato-compile-exec
Nov 17, 2021
64d639b5 make dict of named arrays only once
Nov 24, 2021
957be2f5 PytatoPyOpenCLArrayContext: always invoke MPMS materializer
Nov 25, 2021
c014adbb fix type mismatch in serialize_container argument
Dec 16, 2021
8dab9bc9 PytatoPyOpenCLArrayContext.compile: support returning arrays
Dec 24, 2021
4374e44c test pt_actx.compile for callables returning thawed arrays
Dec 24, 2021
e46e9679 pytato: Convert einsum arg_names into PrefixNamed tags
Dec 26, 2021
8428ca60 mypy cfg: set minimum python version to 3.8 to allow pytato to pass
Dec 26, 2021
5c64c759 Mirgecom downstream CI: Use mpi4py from conda
Dec 26, 2021
1810b0e9 Allow specifying leaf class in recursive map and map-reduce (#128)
Dec 27, 2021
b2e28015 Tighten type information for from_numpy
Jan 11, 2022
9eda169a Disable fail-fast on downstream CI
Jan 11, 2022
427d25c7 Tighten type info on {rec_,}keyed_map_array_container
Jan 11, 2022
8f025148 Actually use _ScalarLike in from_numpy
Jan 12, 2022
e8c74a6c avoid copy in flatten by not calling concatenate
Jan 12, 2022
59b79561 Add a leaf_class parameter to flatten (#142)
Jan 13, 2022
5218b2d6 Remove BaseFakeNumpySpace's dependency on call_loopy (#88)
Jan 16, 2022
3b02d065 add numpy support in each array context to the docs
Jan 16, 2022
474f93a6 remove stray _get_scalar_func_loopy_program
Jan 16, 2022
c88858ec use svg for pypi badge
Jan 16, 2022
83a3da06 fix typo in ndarray serialization
Jan 23, 2022
5964b91f recursively add tags to all arrays in the container
Feb 15, 2022
8f77a2c0 tag pt arrays' axes
Feb 15, 2022
c8427b92 define TaggableCLArray
Feb 15, 2022
91a94fc3 make the frozen type of PytatoPyOpenCLArrayContext to be TaggableCLArrays
Feb 15, 2022
3c4e3048 adds tests specific to PytatoArrayContexts
Feb 15, 2022
b964a1b8 PyOpenCLArrayContext.make_einsum: use `tagged` tags for the output array as well
Feb 15, 2022
458e2eaf Make get_container_context_recursively return actx or error
Mar 01, 2022
6a6be642 fix TaggableCLArray copy with queue
Mar 03, 2022
6b8f1344 Add flat_size_and_dtype
Mar 16, 2022
03f527a6 Stringify return type of flat_size_and_dtype to avoid py3.9+ type subscript
Mar 17, 2022
eef28550 transform DAG before making it a placeholder
Apr 05, 2022
456d8933 allow dataclass containers with only DeviceArrays
Apr 05, 2022
b75ba4f6 add typing to ArrayContext
Apr 07, 2022
2a0006ce with_container_arithmetic: Default cls_has_array_context_attr based on presence of attribute
Apr 19, 2022
17dcc60d support unions in dataclass_array_container
Apr 19, 2022
32dc069e implement _with_new_tags for taggable.Axis
May 01, 2022
cc274f61 adds cla.Array to PytatoPyOpenCLArrayContext.array_types
May 02, 2022
cc795c57 tests to_numpy(...) on frozen arrays
May 02, 2022
21577302 further clarifies the role of ArrayContext.aray_types
May 02, 2022
3b958334 gitlab-ci: remove EXTRA_INSTALL exports
May 12, 2022
6293b32b clean up downstream tests
May 12, 2022
dff68e21 defines JAXArrayContext
May 12, 2022
a45dc761 CI: adds jax to requirements
May 12, 2022
74f7cad9 test JAXArrayContext
May 12, 2022
1cc7f5ea Generalize Pytato Array Context to allow multiple targets
May 12, 2022
47eafd54 Stop setting PY_EXE in Gitlab CI config
May 12, 2022
981a5467 Fix Jax installation on Gitlab CI
May 12, 2022
8d7c872e Drop xfail in test_flatten_array_container
May 20, 2022
4305d758 Deprecate passing unevaluated arrays to compiled functions
Jun 08, 2022
9542cce7 Add folds in arraycontext.impl.pytato
Jun 08, 2022
ec72b093 Log function name in passes of actx.compile
Jun 08, 2022
799e2f19 Pass name of function being compiled to pt.generate_{jax,loopy}
Jun 08, 2022
1f7fff83 Add NameHint tag
Jun 08, 2022
63ef9391 Einsum array argument naming: Use NameHint
Jun 08, 2022
ee0fa5de Rewrite actx.NameHint to pytato.PrefixNamed when tagging
Jun 08, 2022
daa25742 restructure the way 'freeze' is implemented. Move the major bits to `ArrayContext.freeze`
Jun 09, 2022
8b627d73 to_numpy: thaw(freeze(...)) first
Jun 09, 2022
b0d70e67 CI: modernize the freezes on DOFArrays
Jun 09, 2022
13304f68 Simplification: do not exit early out of freeze for 0-size arrays
Jun 09, 2022
37e142e7 Array identifiers in pytato freezes: use "_ary", not "_actx"
Jun 10, 2022
5500b905 Use NameHint/PrefixNamed to generate better kernel names in pytato freeze
Jun 10, 2022
3eac02c9 Add comment to warn against early exits from freeze
Jun 10, 2022
efd2d687 port deprecated uses of freeze and thaw
Jun 11, 2022
e91c52ab Make typing in ArrayContext more precise using TypeVars
Jun 11, 2022
78ee5a1a Add tag_axes
Jun 11, 2022
15fa7788 Add compile_trace_callback to pytato array contexts
Jun 14, 2022
d1157d1a Add type annotationf for _freeze_prg_cache, _dag_transform_cache
Jun 14, 2022
3f57b8be added check for not checking strides of 0-sized arrays
Jun 19, 2022
7f3fc736 Make name of doc build flag more global, explain
Jun 19, 2022
80813d7a filter out unavailable pytest actx factories
Jun 19, 2022
12e173f4 advertise jax support a bit more
Jun 26, 2022
be1429c2 update numpy coverage
Jun 26, 2022
ff1cd0cf rearrange jax.fake_numpy to match other contexts
Jun 26, 2022
b6866767 split dataclass_array_container for easier modification
Jun 27, 2022
5706e44a Support empty containers in `array_equal` (#182)
Jun 29, 2022
a3e68ac9 Expose types for subpackages in package root
Jun 29, 2022
4d8096b8 Bump Python compat target to 3.8
Jun 29, 2022
0569eab0 Clean up array and container type annotations
Jun 29, 2022
9109363d Add ArrayContext.freeze_thaw
Jun 29, 2022
d757876d add some helper functions to taggable_cl_array
Jul 01, 2022
71954a56 pyopencl: add more array container support
Jul 01, 2022
1051d6b6 jax: add more array container support
Jul 01, 2022
70f251f2 pytato: add more array container support
Jul 01, 2022
cda94d4e deprecate to_numpy and from_numpy
Jul 01, 2022
5c9d57ab forward actx.empty_like in actx.np.empty_like
Jul 01, 2022
9f05f723 add container tests to test_array_context_np_like
Jul 01, 2022
fd4d4d8b Look at, silence instances of bugbear B023
Jul 01, 2022
e47d0cf9 Fix fold markers in arraycontext.impl.pytato
Aug 05, 2022
9ab08496 Add folding sections in arraycontext.impl.pytato.compile
Aug 07, 2022
b5feb061 PytatoPyOpenCLArrayContext.freeze: pass allocator to loopy exec
Aug 07, 2022
9d7feaea Pass allocator when converting scalar args to dev buffers
Aug 09, 2022
1593a67b AllocatorInterface renamed to AllocatorBase
Sep 10, 2022
3c9aee68 PytatoPyOpenCLArrayContext: use SVM allocator if available, limit arg size for GPUs (#189)
Sep 19, 2022
dc810902 pytatoactx.compile: support multi-dimensional obj arrays
Sep 19, 2022
167ae7c6 Move ArgSizeLimitingPytatoLoopyPyOpenCLTarget to pytato.utils, remove hard pytato dep
Sep 20, 2022
9e75a457 PytatoPyOpenCLArrayContext: fix non-SVM allocator
Sep 22, 2022
3a5aea7f test the various allocation options
Sep 22, 2022

256 additional commits have been omitted to prevent performance issues.