Commits · b116953895f2c974d5adba9747f2e739b5d03ff1 · Andreas Klöckner / pytato

Browse files

Jun 12, 2022
- distributed/partitioning: preserve axes · b1169538
  Kaushik Kulkarni authored Jun 12, 2022 and Andreas Klöckner committed Jun 12, 2022
  
  b1169538
- oversubscribe: helps distributed tests if the machine doesn't have 2 slots · 26a84053
  Kaushik Kulkarni authored Jun 12, 2022 and Andreas Klöckner committed Jun 12, 2022
  
  26a84053
- Deprecate passing CL devices to pytato codegen · e6357a8e
  Andreas Klöckner authored Jun 12, 2022 and Andreas Klöckner committed Jun 12, 2022
  
  e6357a8e
Jun 08, 2022
- Data wrapper dedup: log to debug, not info · fcd5f54c
  Andreas Klöckner authored Jun 07, 2022 and Andreas Klöckner committed Jun 08, 2022
  
  fcd5f54c
- Stop using @tag_dataclass (mypy doesn't currently understand it) · 89810b21
  Andreas Klöckner authored Jun 07, 2022 and Andreas Klöckner committed Jun 08, 2022
  
  89810b21
- Implement Axis._with_new_tags instead of .copy · 76720eb9
  Andreas Klöckner authored Jun 07, 2022 and Andreas Klöckner committed Jun 08, 2022
  
  76720eb9
Jun 07, 2022
- Allow passing function_name to generate_loopy · bf10a261
  Andreas Klöckner authored Jun 07, 2022 and Andreas Klöckner committed Jun 07, 2022
  
  bf10a261
Jun 01, 2022
- add PYTATO_DEBUG env variable · 638d5ef1
  Matthias Diener authored May 31, 2022 and Andreas Klöckner committed Jun 01, 2022
  
  638d5ef1
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