Commits · 0c1bf0e5fb85c3ee5c87453401604539821e7586 · Andreas Klöckner / pytato · GitLab

Dec 14, 2022
- JAX: Fix DeviceArray for jax 0.4 · 0c1bf0e5
  Matthias Diener authored Dec 14, 2022 and Andreas Klöckner committed Dec 14, 2022
  
  0c1bf0e5
Dec 12, 2022
- distributed example: make it run with arbitrary rank numbers · 49586f65
  Matthias Diener authored Dec 09, 2022 and Andreas Klöckner committed Dec 12, 2022
  
  49586f65
Dec 02, 2022
- [trivial] add comment regarding _DistributedCommReplacer.map_distributed_send (#399) · 03336612
  Matthias Diener authored Dec 02, 2022
  03336612
- small UsersCollector fixes · bc28de87
  Matthias Diener authored Dec 01, 2022 and Andreas Klöckner committed Dec 01, 2022
  
  bc28de87
Nov 23, 2022
- support ConvertibleToShape in make_distributed_recv · 5745fe31
  Matthias Diener authored Nov 23, 2022 and Andreas Klöckner committed Nov 23, 2022
  
  5745fe31
Nov 22, 2022
- Split pytato.distributed up into submodules (no functional changes) · f282b409
  Andreas Klöckner authored Nov 22, 2022 and Andreas Klöckner committed Nov 22, 2022
  
  f282b409
- Test unify_axes_tags · afcc181c
  Kaushik Kulkarni authored May 25, 2022 and Andreas Klöckner committed Nov 21, 2022
  
  afcc181c
- Implement a tag propagation engine · 17ae8c13
  Kaushik Kulkarni authored May 21, 2022 and Andreas Klöckner committed Nov 21, 2022
  
  17ae8c13
- Make all testlib tags derive from the same tag type · d1cb70af
  Kaushik Kulkarni authored May 25, 2022 and Andreas Klöckner committed Nov 21, 2022
  
  d1cb70af
- Add bidict to install requirements · f2b3259a
  Kaushik Kulkarni authored May 25, 2022 and Andreas Klöckner committed Nov 21, 2022
  
  f2b3259a
Nov 18, 2022
- Add sectioning comments in verify_distributed_partition · 6cbf159b
  Andreas Klöckner authored Nov 18, 2022
  
  6cbf159b
Nov 14, 2022
- Use -m mpi4py in MPI test invocations · 30d16d34
  Andreas Klöckner authored Nov 14, 2022 and Andreas Klöckner committed Nov 14, 2022
  
  30d16d34
- Verify DAG partition in random-DAG distributed test · 946b99e9
  Andreas Klöckner authored Nov 09, 2022 and Andreas Klöckner committed Nov 13, 2022
  
  946b99e9
- Test verify_distributed_partition · f47b4f15
  Matthias Diener authored Nov 09, 2022 and Andreas Klöckner committed Nov 13, 2022
  
  f47b4f15
- Add verify_distributed_partition · 744a40ad
  Andreas Klöckner authored Nov 09, 2022 and Andreas Klöckner committed Nov 13, 2022
  744a40ad
- test_distributed: avoid separate distributed import · 829ea7d2
  Andreas Klöckner authored Nov 09, 2022 and Andreas Klöckner committed Nov 13, 2022
  
  829ea7d2
- Fix sectioning, improve naming in pt.distributed · 68a38315
  Andreas Klöckner authored Nov 09, 2022 and Andreas Klöckner committed Nov 13, 2022
  
  68a38315
- Fix typo: _map_distributed_graph_partition_nodes · 9eb7358f
  Matthias Diener authored Nov 09, 2022 and Andreas Klöckner committed Nov 13, 2022
  
  9eb7358f
- Add some assertions in number_distributed_tags and _post_receive · e3d9ef52
  Matthias Diener authored Nov 09, 2022 and Andreas Klöckner committed Nov 13, 2022
  
  e3d9ef52
- Use MPI4py types in type annotation in pt.distributed · acc934ca
  Matthias Diener authored Nov 09, 2022 and Andreas Klöckner committed Nov 13, 2022
  
  acc934ca
- Remove stray distributed docs from pytato.partition · 2435e04d
  Andreas Klöckner authored Oct 31, 2022 and Andreas Klöckner committed Nov 13, 2022
  
  2435e04d
- add mpi4py intersphinx · 21abd21b
  Matthias Diener authored Oct 31, 2022 and Andreas Klöckner committed Nov 13, 2022
  
  21abd21b
Nov 09, 2022
- Add regressions for CachedWalkMapper with extra args · 2d944726
  Kaushik Kulkarni authored Nov 07, 2022 and Andreas Klöckner committed Nov 09, 2022
  
  2d944726
- Pass extra `args`, `kwargs` to CachedWalkMapper's mapper methods · 96cdbc17
  Kaushik Kulkarni authored Nov 07, 2022 and Andreas Klöckner committed Nov 09, 2022
  96cdbc17
- Rename variables in transform to avoid collision with other modules · b76d4f97
  Kaushik Kulkarni authored Nov 09, 2022 and Andreas Klöckner committed Nov 09, 2022
  
  b76d4f97
- Rename cache_key -> get_cache_key · 5346f7c4
  Kaushik Kulkarni authored Nov 09, 2022 and Andreas Klöckner committed Nov 09, 2022
  5346f7c4
- test lower_to_index_lambda · 914ba7c4
  Kaushik Kulkarni authored Nov 06, 2022 and Andreas Klöckner committed Nov 09, 2022
  
  914ba7c4
- refactor: make lower_to_index_lambda a mapper · 9f09e490
  Kaushik Kulkarni authored Nov 06, 2022 and Andreas Klöckner committed Nov 09, 2022
  
  9f09e490
- [mypy]: mypy-0.990 caught an incorrect subtype method signature · 8947983e
  Kaushik Kulkarni authored Nov 07, 2022
  
  8947983e
Nov 06, 2022
- pyrsistent -> immutables · 44200cd9
  Kaushik Kulkarni authored Nov 06, 2022 and Andreas Klöckner committed Nov 06, 2022
  
  44200cd9
- Fix return type annotation of Axis._with_new_tag · fbb1d824
  Andreas Klöckner authored Nov 06, 2022 and Andreas Klöckner committed Nov 06, 2022
  
  fbb1d824
Nov 05, 2022
- adds check for AbstractResultWithNamedArrays.__eq__ · 31a1c2ea
  Kaushik Kulkarni authored Nov 04, 2022 and Andreas Klöckner committed Nov 05, 2022
  
  31a1c2ea
- ignore immutables in docs · c8277c75
  Kaushik Kulkarni authored Nov 04, 2022 and Andreas Klöckner committed Nov 05, 2022
  
  c8277c75
Nov 04, 2022
- lower index lambda to index lambda · 156f2874
  Kaushik Kulkarni authored Nov 04, 2022
  
  156f2874
Nov 02, 2022
- implements lower_to_index_lambda · 23579cff
  Kaushik Kulkarni authored Nov 01, 2022
  
  23579cff
- make IndexLambda.bindings immutable · e7d418c5
  Kaushik Kulkarni authored Nov 01, 2022
  
  e7d418c5
- skip asciidag test if package not available · 6b01cb18
  Matthias Diener authored Nov 02, 2022
  
  6b01cb18
- pyrsistent.pmap -> immutables.Map; adds an assert to ensure the linear scheduling · 689b50e3
  Kaushik Kulkarni authored Nov 02, 2022
  
  689b50e3
- use different comm_tag for different receives · f3cd7bc1
  Kaushik Kulkarni authored Oct 31, 2022
  
  f3cd7bc1
Oct 29, 2022
- adds expand_dims, ExpandedDimsReshapeTag to docs · 6b6d9f3e
  Kaushik Kulkarni authored May 25, 2022
  
  6b6d9f3e