Andreas Klöckner authored Jun 28, 2024 and Andreas Klöckner committed Jul 01, 2024

Closes gh-509

dependabot[bot] authored Apr 13, 2024 and Andreas Klöckner committed Apr 13, 2024

Bumps [actions/checkout](https://github.com/actions/checkout) from 3 to 4.
- [Release notes](https://github.com/actions/checkout/releases)
- [Changelog](https://github.com/actions/checkout/blob/main/CHANGELOG.md)
- [Commits](https://github.com/actions/checkout/compare/v3...v4

)

---
updated-dependencies:
- dependency-name: actions/checkout
  dependency-type: direct:production
  update-type: version-update:semver-major
...

Signed-off-by: dependabot[bot] <support@github.com>

* Add tag to store array creation traceback

* don't make it a unique tag

* adds a common _get_default_tags

* Change back to UniqueTag

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

* use _get_default_tags

* store a tupleized StackSummary

* work around mypy

* more line fixes

* use a class for the traceback instead of tuples

* also test to_stacksummary

* flake8

* Add remove_tags_of_type

* test_array_dot_repr: Remove CreatedAt tags before comparing

* only add CreatedAt in debug mode

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

* restructure test_created_at

* make _PytatoStackSummary a dataclass

* add __repr__

* fix 2 tests

* illustrate test failure with construct_intestine_graph

* shorten traceback printing

* use separate field for CreatedAt

* fix tests

* fix doctest

* make it a tag again

* use tooltip instead of table row

* force openmpi usage

* check for existing CreatedAt and make it a UniqueTag again

* flake8

* add simple equality test

* lint fixes

* add InfoTag class and filter tags based on it

* fix doc

* another doc fix

* use IgnoredForEqualityTag

* UNDO BEFORE MERGE: use external project branches

* Revert "UNDO BEFORE MERGE: use external project branches"

This reverts commit 71dd791b45cee918d33ac8c4cb8e9a3cf643ec37.

* Revert "use IgnoredForEqualityTag"

This reverts commit cd67d684d1c777bc3efef080031446e033e75f42.

* rename InfoTag -> IgnoredForEqualityTag

* more stringent tests

* undo unnecessary test changes

* Revert "Revert "use IgnoredForEqualityTag""

This reverts commit bfb22ba66d0872c3848a4ab41260f9f82ffa188d.

* simplify condition

* Revert "simplify condition"

This reverts commit 1111b7915e23561161eee132e8d5ccb716b101eb.

* bump pytools version + a few spelling fixes

* remove duplicated self.axes in hash()

* use Taggable{__eq__,__hash__}

* add another test

* add vis test

* make _PytatoFrameSummary, _PytatoStackSummary undocumented

* use Taggable.__hash__ for tags in Array.__hash__

* change dataclass to attrs

* flake8

* Taggable.__eq__

* add Array.tagged()

* restrict to DEBUG_ENABLED

* force DEBUG_ENABLED for test

* work around mypy/attrs issue

* fix for fields

* undo mypy ignores

* rewrite to use a new field in Array, non_equality_tags

* misc fixes

* undo some unecessary changes

* more misc fixes

* copymapper, tests

* explicitly enable/disable traceback

* add hash test

* undo more unnecessary changes

* more lint fixes

* run all examples, fix demo_distributed_node_duplication

* enable CreatedAt for distributed nodes

* add support for make_distributed_send_ref_holder

* add to MPMSMaterializer

* Get precise traceback of array creation (#480)

* rename enable_traceback_tag

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

* remove update_persistent_hash

* improve doc of _PytatoFrameSummary, _PytatoStackSummary

* move _Pytato{Frame,Stack}Summary to tags, convert to dataclass

* add AxisPermutation CreatedAt

* fix doc

* make _get_created_at_tag return a frozenset

---------

Co-authored-by: Kaushik Kulkarni <kaushikcfd@gmail.com>
Co-authored-by: Andreas Klöckner <inform@tiker.net>
Co-authored-by: Matt Smith <mjsmith6@illinois.edu>

* Update to pass *args and **kwargs through

* We should also pass them through in map_non_contiguous_advanced_index

Andreas Klöckner authored Mar 01, 2024 and Andreas Klöckner committed Mar 01, 2024

The latest version of jax updated the location of the config object:
https://jax.readthedocs.io/en/latest/changelog.html#jax-0-4-25-feb-26-2024

Should work with older versions, has been deprecated a while.

h/t @alexfikl via https://github.com/inducer/arraycontext/pull/252

* fix some warnings in tests

* add comment+complex/real arange test

* number_distributed_tags: non-set, non-sorted numbering

* make the test a bit more difficult

* improve test

* add comments based on review

* Revert sorting of symbolic tags

Sorting fails when symbolic tags contain bare classes (which can not be
compared most of the time).

The attached test case fails without this PR.

* undo doc

* run all examples, fix demo_distributed_node_duplication

* remove premature change

* call super().__init__() in mappers

* fixes for functions

* add update_from_callee_clone function to update parent mapper's side effect data

* fix _DistributedInputReplacer

* failed attempt at tweaking MPMSMaterializer to avoid type checking issue

if rec is supposed to call map_call, map_call should return an MPMSMaterializer

* remove implementation of DirectPredecessorsGetter.map_named_call_result

* back out some placeholder stuff

* remove implementation of LocalSendRecvDepGatherer.map_named_call_result

* fix arg checking in FunctionDefinition.__call__

* forget about functions in MPMSMaterializer for now

* handle results from non-inlined calls correctly in Inliner

* remove update_from_callee_clone

* move CodeGenPreprocessor function error to map_named_call_result

* remove unused mypy ignore

Andreas Klöckner authored Nov 21, 2023 and Andreas Klöckner committed Nov 21, 2023

https://github.com/python/mypy/issues/16499