Commits · ea06b523fc7dcb64db8a809249073e2d25ece01d · Andreas Klöckner / pytato · GitLab

Apr 05, 2023
- test pytato.pad · ea06b523
  Kaushik Kulkarni authored Mar 12, 2023 and Andreas Klöckner committed Apr 04, 2023
  
  ea06b523
- Implement pytato.pad · ea909d3d
  Kaushik Kulkarni authored Mar 12, 2023 and Andreas Klöckner committed Apr 04, 2023
  
  ea909d3d
Mar 14, 2023
- Test NamedArrays for JAXTarget · 1f38f45a
  Kaushik Kulkarni authored Mar 14, 2023
  
  1f38f45a
- support generating JAX code with NamedArrays in expr graph · e7c35b09
  Kaushik Kulkarni authored Mar 14, 2023
  
  e7c35b09
- make get_dot_graph_from_partition work with duplicated computation · a00f8b4f
  Matt Smith authored Jan 30, 2023 and Andreas Klöckner committed Mar 14, 2023
  
  a00f8b4f
Mar 06, 2023
- Avoid using non-export jax.numpy.ndarray (mypy complaint) · cf3673a1
  Andreas Klöckner authored Mar 06, 2023 and Andreas Klöckner committed Mar 06, 2023
  
  cf3673a1
Feb 17, 2023
- check that outputs tagged with ImplStored do not result in additional global temps · f0f76312
  Kaushik Kulkarni authored Feb 01, 2023 and Andreas Klöckner committed Feb 17, 2023
  
  f0f76312
- remove ImplStored tags on outputs · 5c3884df
  Kaushik Kulkarni authored Feb 01, 2023 and Andreas Klöckner committed Feb 17, 2023
  
  5c3884df
Jan 19, 2023
- Add regression for rewrite_einsums_with_no_broadcasts · bcb9f4bc
  Kaushik Kulkarni authored Jan 18, 2023 and Andreas Klöckner committed Jan 18, 2023
  
  bcb9f4bc
- Add pytato.rewrite_einsums_with_no_broadcasts transformation · dfae6813
  Kaushik Kulkarni authored Jan 18, 2023 and Andreas Klöckner committed Jan 18, 2023
  
  dfae6813
Jan 17, 2023
- Test lowering to substitution rule invocation in loopy · fe87c94e
  Kaushik Kulkarni authored Jan 13, 2023 and Andreas Klöckner committed Jan 16, 2023
  
  fe87c94e
- Implement pytato.target.loopy.ImplSubstitution · c5164d00
  Kaushik Kulkarni authored Jan 13, 2023 and Andreas Klöckner committed Jan 16, 2023
  
  c5164d00
Jan 10, 2023
- use make_dict_of_named_arrays instead of constructor · 01f8d122
  Matthias Diener authored Jan 10, 2023 and Andreas Klöckner committed Jan 10, 2023
  
  01f8d122
Jan 09, 2023
- Make flake8-comprehensions happy · b26d4fcf
  Andreas Klöckner authored Jan 09, 2023
  
  b26d4fcf
Dec 27, 2022
- Minor distributed visualization improvements (#385) · 26b28284
  Matthias Diener authored Dec 27, 2022
  26b28284
- Remove unused reverse_graph (#376) · d82ada25
  Matthias Diener authored Dec 27, 2022
  d82ada25
- Track move of show_dot to pytools.graphviz · 379ee077
  Andreas Klöckner authored Dec 27, 2022 and Andreas Klöckner committed Dec 27, 2022
  
  379ee077
Dec 25, 2022
- Sphinx nitpick ignore disappeared :mod:`attrs` · 5aee209c
  Andreas Klöckner authored Dec 24, 2022 and Andreas Klöckner committed Dec 24, 2022
  
  5aee209c
- Swap bool8 -> bool_ to placate numpy deprecation · f8238135
  Andreas Klöckner authored Dec 24, 2022 and Andreas Klöckner committed Dec 24, 2022
  
  f8238135
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