Commits · 644eacbe13d03323756f49edfb69b090502a12d7 · Andreas Klöckner / pytato

Browse files

May 25, 2022
- Mapper: allow 'expr' to be subclasses of ArrayOrNames · 644eacbe
  Kaushik Kulkarni authored May 23, 2022 and Andreas Klöckner committed May 25, 2022
  
  644eacbe
May 24, 2022
- implements expand_dims · a59d3d34
  Kaushik Kulkarni authored May 21, 2022 and Andreas Klöckner committed May 23, 2022
  a59d3d34
- generalize the types of Reshape · 02492f5f
  Kaushik Kulkarni authored May 21, 2022 and Andreas Klöckner committed May 23, 2022
  
  02492f5f
May 23, 2022
- [mypy] account for a more precise np.isnan · 80b6dab9
  Kaushik Kulkarni authored May 21, 2022
  
  80b6dab9
- [pytato.distributed]allow only evaluated shape values before allocating a buffer · e21ab5fd
  Kaushik Kulkarni authored May 21, 2022
  
  e21ab5fd
May 19, 2022
- check identity in EqualityComparisonMapper · 710baffb
  Kaushik Kulkarni authored May 19, 2022
  710baffb
May 11, 2022
- Do not emit statements or domains for empty arrays · 57afae65
  Andreas Klöckner authored May 10, 2022 and Andreas Klöckner committed May 10, 2022
  
  57afae65
May 08, 2022
- move transform.py to transform/__init__.py · faee1d2f
  Kaushik Kulkarni authored May 08, 2022
  
  faee1d2f
May 04, 2022
- ci:pylint: remove loopy branch change · 2785a6dc
  Alexandru Fikl authored May 04, 2022 and Andreas Klöckner committed May 04, 2022
  
  2785a6dc
- run-mypy: friendlier venv support · 075a7f8c
  Alexandru Fikl authored May 04, 2022 and Andreas Klöckner committed May 04, 2022
  
  075a7f8c
May 03, 2022
- Install jax and pygments for Gitlab CI · c77a7052
  Andreas Klöckner authored May 03, 2022 and Andreas Klöckner committed May 03, 2022
  
  c77a7052
- Tests JAXPythonTarget · bec27777
  Kaushik Kulkarni authored Dec 17, 2021 and Andreas Klöckner committed May 03, 2022
  
  bec27777
- [mypy]: ignore pygments-specific routines · ad2e67ea
  Kaushik Kulkarni authored Dec 17, 2021 and Andreas Klöckner committed May 03, 2022
  
  ad2e67ea
- adds JAX-specific bits to docs · e67c6ee2
  Kaushik Kulkarni authored Dec 17, 2021 and Andreas Klöckner committed May 03, 2022
  
  e67c6ee2
- Implements JAX Target · 007c30ce
  Kaushik Kulkarni authored Dec 07, 2021 and Andreas Klöckner committed May 03, 2022
  
  007c30ce
- cleanup: do not enforce Target.bind_program · 545a72bb
  Kaushik Kulkarni authored May 02, 2022 and Andreas Klöckner committed May 03, 2022
  
  545a72bb
- Doc fix · 01ea3149
  nchristensen authored May 03, 2022 and Andreas Klöckner committed May 03, 2022
  01ea3149
- test advanced indexing into zero-long axes · e88cac27
  Kaushik Kulkarni authored May 02, 2022 and Andreas Klöckner committed May 02, 2022
  
  e88cac27
- fixes a bug in _is_non_negative · 7be6758d
  Kaushik Kulkarni authored May 02, 2022 and Andreas Klöckner committed May 02, 2022
  
  7be6758d
- make indexing into non-positive dimension axes illegal · 9bf14e4d
  Kaushik Kulkarni authored May 02, 2022 and Andreas Klöckner committed May 02, 2022
  
  9bf14e4d
Apr 29, 2022
- add float to scalar types · ea541dac
  Kaushik Kulkarni authored Apr 28, 2022 and Andreas Klöckner committed Apr 29, 2022
  ea541dac
- index_lambda_to_hlo: support NaNs · ee563aae
  Kaushik Kulkarni authored Apr 28, 2022 and Andreas Klöckner committed Apr 29, 2022
  
  ee563aae
- mypy ignore typeshed-issue-7739 · 735a8225
  Kaushik Kulkarni authored Apr 28, 2022 and Andreas Klöckner committed Apr 28, 2022
  
  735a8225
Apr 24, 2022
- tests if pytato-DAG<with-comm-nodes> -> partitioned DAGs is reproducible · f21e6ab7
  Kaushik Kulkarni authored Apr 22, 2022
  
  f21e6ab7
- tests if pytato-DAG -> loopy is reproducible · 98921d9c
  Kaushik Kulkarni authored Apr 22, 2022
  
  98921d9c
- add helpers for generating random DAGs · 264099dc
  Kaushik Kulkarni authored Apr 24, 2022
  
  264099dc
- test pickling support for DAGs · cc4b0391
  Kaushik Kulkarni authored Mar 24, 2022 and Andreas Klöckner committed Apr 23, 2022
  
  cc4b0391
- Remove np.nan from expression trees · e1aee585
  Kaushik Kulkarni authored Mar 24, 2022 and Andreas Klöckner committed Apr 23, 2022
  e1aee585
Apr 19, 2022
- Remove PartIDTag from nodes in partition (closes gh-307) · d015f914
  Andreas Klöckner authored Apr 18, 2022 and Andreas Klöckner committed Apr 19, 2022
  
  d015f914
- Implement _with_new_tags in DistributedSend · 8664b871
  Andreas Klöckner authored Apr 18, 2022 and Andreas Klöckner committed Apr 19, 2022
  
  8664b871
- Remove unused iteration variable in number_distributed_tags · b8e351ba
  Andreas Klöckner authored Apr 18, 2022 and Andreas Klöckner committed Apr 19, 2022
  
  b8e351ba
- Minor doc/comment improvements in partition · 34582cc5
  Andreas Klöckner authored Apr 18, 2022 and Andreas Klöckner committed Apr 19, 2022
  
  34582cc5
Apr 13, 2022
- Remove uses of newly-deprecated symbols in pytools.tag · aab74009
  Andreas Klöckner authored Apr 12, 2022 and Andreas Klöckner committed Apr 12, 2022
  
  aab74009
Apr 07, 2022
- add PooledBuffer to dedup cache key (#304) · 185aeafb
  Matthias Diener authored Apr 07, 2022
  185aeafb
Apr 06, 2022
- adds a regression on a DAG with no comm. nodes · 4bdef631
  Kaushik Kulkarni authored Apr 01, 2022 and Andreas Klöckner committed Apr 05, 2022
  
  4bdef631
- materialize nodes before find_distributed_partition · 208ed3c5
  Kaushik Kulkarni authored Mar 20, 2022 and Andreas Klöckner committed Apr 05, 2022
  
  208ed3c5
- Adds example to demonstrate eager send scheduling · 96282ec9
  Kaushik Kulkarni authored Mar 20, 2022 and Andreas Klöckner committed Apr 05, 2022
  
  96282ec9
- find_distributed_partition: better algorithm · acfd5d30
  Kaushik Kulkarni authored Mar 20, 2022 and Andreas Klöckner committed Apr 05, 2022
  acfd5d30
- implements CopyMapperWithExtraArgs · f36cf325
  Kaushik Kulkarni authored Mar 20, 2022 and Andreas Klöckner committed Apr 05, 2022
  
  f36cf325
- Implements DirectPredecessorsGetter · 5455ac7a
  Kaushik Kulkarni authored Mar 19, 2022 and Andreas Klöckner committed Apr 05, 2022
  
  5455ac7a