Commits · d1157d1aed6e9a2c713714c656843c260e7bfa48 · Kaushik Kulkarni / arraycontext · GitLab

Jun 14, 2022
- Add type annotationf for _freeze_prg_cache, _dag_transform_cache · d1157d1a
  Andreas Klöckner authored Jun 14, 2022 and Andreas Klöckner committed Jun 14, 2022
  
  d1157d1a
- Add compile_trace_callback to pytato array contexts · 15fa7788
  Andreas Klöckner authored Jun 14, 2022 and Andreas Klöckner committed Jun 14, 2022
  
  15fa7788
Jun 11, 2022
- Add tag_axes · 78ee5a1a
  Andreas Klöckner authored Jun 10, 2022 and Andreas Klöckner committed Jun 11, 2022
  78ee5a1a
- Make typing in ArrayContext more precise using TypeVars · e91c52ab
  Andreas Klöckner authored Jun 10, 2022 and Andreas Klöckner committed Jun 11, 2022
  
  e91c52ab
- port deprecated uses of freeze and thaw · efd2d687
  Alexandru Fikl authored Jun 11, 2022 and Andreas Klöckner committed Jun 11, 2022
  
  efd2d687
Jun 10, 2022
- Add comment to warn against early exits from freeze · 3eac02c9
  Andreas Klöckner authored Jun 10, 2022
  
  3eac02c9
- Use NameHint/PrefixNamed to generate better kernel names in pytato freeze · 5500b905
  Andreas Klöckner authored Jun 07, 2022 and Andreas Klöckner committed Jun 10, 2022
  
  5500b905
- Array identifiers in pytato freezes: use "_ary", not "_actx" · 37e142e7
  Andreas Klöckner authored Jun 10, 2022 and Andreas Klöckner committed Jun 10, 2022
  
  37e142e7
- Simplification: do not exit early out of freeze for 0-size arrays · 13304f68
  Kaushik Kulkarni authored Jun 09, 2022 and Andreas Klöckner committed Jun 09, 2022
  13304f68
Jun 09, 2022
- CI: modernize the freezes on DOFArrays · b0d70e67
  Kaushik Kulkarni authored May 09, 2022 and Andreas Klöckner committed Jun 09, 2022
  
  b0d70e67
- to_numpy: thaw(freeze(...)) first · 8b627d73
  Kaushik Kulkarni authored May 01, 2022 and Andreas Klöckner committed Jun 09, 2022
  
  8b627d73
- restructure the way 'freeze' is implemented. Move the major bits to `ArrayContext.freeze` · daa25742
  Kaushik Kulkarni authored May 01, 2022 and Andreas Klöckner committed Jun 09, 2022
  daa25742
Jun 08, 2022
- Rewrite actx.NameHint to pytato.PrefixNamed when tagging · ee0fa5de
  Andreas Klöckner authored Jun 07, 2022 and Andreas Klöckner committed Jun 08, 2022
  
  ee0fa5de
- Einsum array argument naming: Use NameHint · 63ef9391
  Andreas Klöckner authored Jun 07, 2022 and Andreas Klöckner committed Jun 08, 2022
  
  63ef9391
- Add NameHint tag · 1f7fff83
  Andreas Klöckner authored Jun 07, 2022 and Andreas Klöckner committed Jun 08, 2022
  
  1f7fff83
- Pass name of function being compiled to pt.generate_{jax,loopy} · 799e2f19
  Andreas Klöckner authored Jun 07, 2022 and Andreas Klöckner committed Jun 08, 2022
  
  799e2f19
- Log function name in passes of actx.compile · ec72b093
  Andreas Klöckner authored Jun 07, 2022 and Andreas Klöckner committed Jun 08, 2022
  
  ec72b093
- Add folds in arraycontext.impl.pytato · 9542cce7
  Andreas Klöckner authored Jun 07, 2022 and Andreas Klöckner committed Jun 08, 2022
  
  9542cce7
- Deprecate passing unevaluated arrays to compiled functions · 4305d758
  Andreas Klöckner authored Jun 07, 2022 and Andreas Klöckner committed Jun 08, 2022
  
  4305d758
May 20, 2022
- Drop xfail in test_flatten_array_container · 8d7c872e
  Andreas Klöckner authored May 19, 2022 and Andreas Klöckner committed May 20, 2022
  8d7c872e
May 13, 2022
- Fix Jax installation on Gitlab CI · 981a5467
  Andreas Klöckner authored May 12, 2022
  
  981a5467
- Stop setting PY_EXE in Gitlab CI config · 47eafd54
  Andreas Klöckner authored May 12, 2022
  
  47eafd54
May 12, 2022
- Generalize Pytato Array Context to allow multiple targets · 1cc7f5ea
  Kaushik Kulkarni authored Jan 21, 2022
  1cc7f5ea
- test JAXArrayContext · 74f7cad9
  Kaushik Kulkarni authored Sep 19, 2021
  
  74f7cad9
- CI: adds jax to requirements · a45dc761
  Kaushik Kulkarni authored Jul 18, 2021
  
  a45dc761
- defines JAXArrayContext · dff68e21
  Kaushik Kulkarni authored Jul 18, 2021
  dff68e21
- clean up downstream tests · 6293b32b
  Alexandru Fikl authored May 10, 2022 and Andreas Klöckner committed May 12, 2022
  
  6293b32b
- gitlab-ci: remove EXTRA_INSTALL exports · 3b958334
  Alexandru Fikl authored May 10, 2022 and Andreas Klöckner committed May 12, 2022
  
  3b958334
May 02, 2022
- further clarifies the role of ArrayContext.aray_types · 21577302
  Kaushik Kulkarni authored May 01, 2022 and Andreas Klöckner committed May 02, 2022
  
  21577302
- tests to_numpy(...) on frozen arrays · cc795c57
  Kaushik Kulkarni authored May 01, 2022 and Andreas Klöckner committed May 02, 2022
  
  cc795c57
- adds cla.Array to PytatoPyOpenCLArrayContext.array_types · cc274f61
  Kaushik Kulkarni authored May 01, 2022 and Andreas Klöckner committed May 02, 2022
  
  cc274f61
- implement _with_new_tags for taggable.Axis · 32dc069e
  Alexandru Fikl authored Apr 30, 2022 and Andreas Klöckner committed May 01, 2022
  
  32dc069e
Apr 20, 2022
- support unions in dataclass_array_container · 17dcc60d
  Alexandru Fikl authored Apr 19, 2022 and Andreas Klöckner committed Apr 19, 2022
  
  17dcc60d
- with_container_arithmetic: Default cls_has_array_context_attr based on presence of attribute · 2a0006ce
  Andreas Klöckner authored Apr 13, 2022 and Andreas Klöckner committed Apr 19, 2022
  
  2a0006ce
Apr 07, 2022
- add typing to ArrayContext · b75ba4f6
  Alexandru Fikl authored Apr 05, 2022 and Andreas Klöckner committed Apr 07, 2022
  
  b75ba4f6
Apr 06, 2022
- allow dataclass containers with only DeviceArrays · 456d8933
  Alexandru Fikl authored Apr 05, 2022 and Andreas Klöckner committed Apr 05, 2022
  
  456d8933
- transform DAG before making it a placeholder · eef28550
  Kaushik Kulkarni authored Mar 31, 2022 and Andreas Klöckner committed Apr 05, 2022
  eef28550
Mar 17, 2022
- Stringify return type of flat_size_and_dtype to avoid py3.9+ type subscript · 03f527a6
  Andreas Klöckner authored Mar 17, 2022 and Andreas Klöckner committed Mar 17, 2022
  
  03f527a6
- Add flat_size_and_dtype · 6b8f1344
  Andreas Klöckner authored Mar 16, 2022 and Andreas Klöckner committed Mar 16, 2022
  6b8f1344
Mar 03, 2022
- fix TaggableCLArray copy with queue · 6a6be642
  Alexandru Fikl authored Feb 19, 2022 and Andreas Klöckner committed Mar 03, 2022
  
  6a6be642