Commits · 78ee5a1ae3d1f3afe1809023dc3377f4e98960ac · Andreas Klöckner / arraycontext · GitLab

Snippets Groups Projects

Jun 11, 2022
- Add tag_axes · 78ee5a1a
  Andreas Klöckner authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  Co-authored-by: Kaushik Kulkarni <kaushikcfd@gmail.com>
  78ee5a1a
- Make typing in ArrayContext more precise using TypeVars · e91c52ab
  Andreas Klöckner authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  e91c52ab
- port deprecated uses of freeze and thaw · efd2d687
  Alexandru Fikl authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  efd2d687
Jun 10, 2022
- Add comment to warn against early exits from freeze · 3eac02c9
  Andreas Klöckner authored 2 years ago
  
  3eac02c9
- Use NameHint/PrefixNamed to generate better kernel names in pytato freeze · 5500b905
  Andreas Klöckner authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  5500b905
- Array identifiers in pytato freezes: use "_ary", not "_actx" · 37e142e7
  Andreas Klöckner authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  37e142e7
- Simplification: do not exit early out of freeze for 0-size arrays · 13304f68
  Kaushik Kulkarni authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  Co-authored-by: Matthew Smith <mjsmith6@illinois.edu>
  13304f68
Jun 09, 2022
- CI: modernize the freezes on DOFArrays · b0d70e67
  Kaushik Kulkarni authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  b0d70e67
- to_numpy: thaw(freeze(...)) first · 8b627d73
  Kaushik Kulkarni authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  8b627d73
- restructure the way 'freeze' is implemented. Move the major bits to `ArrayContext.freeze` · daa25742
  Kaushik Kulkarni authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  Co-authored-by: Andreas Kloeckner <inform@tiker.net>
  daa25742
Jun 08, 2022
- Rewrite actx.NameHint to pytato.PrefixNamed when tagging · ee0fa5de
  Andreas Klöckner authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  ee0fa5de
- Einsum array argument naming: Use NameHint · 63ef9391
  Andreas Klöckner authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  63ef9391
- Add NameHint tag · 1f7fff83
  Andreas Klöckner authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  1f7fff83
- Pass name of function being compiled to pt.generate_{jax,loopy} · 799e2f19
  Andreas Klöckner authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  799e2f19
- Log function name in passes of actx.compile · ec72b093
  Andreas Klöckner authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  ec72b093
- Add folds in arraycontext.impl.pytato · 9542cce7
  Andreas Klöckner authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  9542cce7
- Deprecate passing unevaluated arrays to compiled functions · 4305d758
  Andreas Klöckner authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  4305d758
May 20, 2022
- Drop xfail in test_flatten_array_container · 8d7c872e
  Andreas Klöckner authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  Needs https://github.com/inducer/loopy/pull/624.
  8d7c872e
May 13, 2022
- Fix Jax installation on Gitlab CI · 981a5467
  Andreas Klöckner authored 2 years ago
  
  981a5467
- Stop setting PY_EXE in Gitlab CI config · 47eafd54
  Andreas Klöckner authored 2 years ago
  
  47eafd54
May 12, 2022
- Generalize Pytato Array Context to allow multiple targets · 1cc7f5ea
  Kaushik Kulkarni authored 3 years ago
  
  Also implements PytatoJAXTarget. Co-authored-by: Alexandru Fikl <alexfikl@gmail.com>
  1cc7f5ea
- test JAXArrayContext · 74f7cad9
  Kaushik Kulkarni authored 3 years ago
  
  74f7cad9
- CI: adds jax to requirements · a45dc761
  Kaushik Kulkarni authored 3 years ago
  
  a45dc761
- defines JAXArrayContext · dff68e21
  Kaushik Kulkarni authored 3 years ago
  
  Co-authored-by: Alexandru Fikl <alexfikl@gmail.com>
  dff68e21
- clean up downstream tests · 6293b32b
  Alexandru Fikl authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  6293b32b
- gitlab-ci: remove EXTRA_INSTALL exports · 3b958334
  Alexandru Fikl authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  3b958334
May 02, 2022
- further clarifies the role of ArrayContext.aray_types · 21577302
  Kaushik Kulkarni authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  21577302
- tests to_numpy(...) on frozen arrays · cc795c57
  Kaushik Kulkarni authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  cc795c57
- adds cla.Array to PytatoPyOpenCLArrayContext.array_types · cc274f61
  Kaushik Kulkarni authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  cc274f61
- implement _with_new_tags for taggable.Axis · 32dc069e
  Alexandru Fikl authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  32dc069e
Apr 20, 2022
- support unions in dataclass_array_container · 17dcc60d
  Alexandru Fikl authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  17dcc60d
- with_container_arithmetic: Default cls_has_array_context_attr based on presence of attribute · 2a0006ce
  Andreas Klöckner authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  2a0006ce
Apr 07, 2022
- add typing to ArrayContext · b75ba4f6
  Alexandru Fikl authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  b75ba4f6
Apr 06, 2022
- allow dataclass containers with only DeviceArrays · 456d8933
  Alexandru Fikl authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  456d8933
- transform DAG before making it a placeholder · eef28550
  Kaushik Kulkarni authored 2 years ago and Andreas Klöckner committed 2 years ago
  
  Co-authored-by: Andreas Kloeckner <inform@tiker.net>
  eef28550
Mar 17, 2022
- Stringify return type of flat_size_and_dtype to avoid py3.9+ type subscript · 03f527a6
  Andreas Klöckner authored 3 years ago and Andreas Klöckner committed 3 years ago
  
  03f527a6
- Add flat_size_and_dtype · 6b8f1344
  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>
  6b8f1344
Mar 03, 2022
- fix TaggableCLArray copy with queue · 6a6be642
  Alexandru Fikl authored 3 years ago and Andreas Klöckner committed 3 years ago
  
  6a6be642
Mar 02, 2022
- Make get_container_context_recursively return actx or error · 458e2eaf
  Andreas Klöckner authored 3 years ago and Andreas Klöckner committed 3 years ago
  
  458e2eaf
Feb 15, 2022
- PyOpenCLArrayContext.make_einsum: use `tagged` tags for the output array as well · b964a1b8
  Kaushik Kulkarni authored 3 years ago and Andreas Klöckner committed 3 years ago
  
  b964a1b8