Commits · b6dcd1b55d9dbb4492c32440389fd9659fb334ca · Andreas Klöckner / pytato

Browse files

Apr 21, 2023
- Type cache key as hashable · b6dcd1b5
  Andreas Klöckner authored Apr 20, 2023
  
  b6dcd1b5
Apr 20, 2023
- improve precision of map_and_copy's type · 668020de
  Kaushik Kulkarni authored Apr 19, 2023
  
  668020de
Apr 19, 2023
- make the parametric type of CachedMapper to be the return type · f46d9bb0
  Kaushik Kulkarni authored Apr 18, 2023 and Andreas Klöckner committed Apr 19, 2023
  
  f46d9bb0
- corrects docs about NormalizedSlice · ddf8e8cf
  Kaushik Kulkarni authored Apr 16, 2023 and Andreas Klöckner committed Apr 19, 2023
  ddf8e8cf
- Add `axis` argument to pytato.squeeze · 6e423050
  Kaushik Kulkarni authored Apr 17, 2023 and Andreas Klöckner committed Apr 19, 2023
  
  6e423050
Apr 17, 2023
- Fix some instances of flake8-comprehensions C419 · e45d63b5
  Andreas Klöckner authored Apr 16, 2023 and Andreas Klöckner committed Apr 16, 2023
  
  e45d63b5
Apr 10, 2023
- Add DistributedRecv cast in distributed.partition for mypy 1.2 · 3083e95f
  Andreas Klöckner authored Apr 09, 2023 and Andreas Klöckner committed Apr 09, 2023
  
  3083e95f
Apr 05, 2023
- Install pytest for Gitlab examples CI · 8e786f05
  Andreas Klöckner authored Apr 05, 2023
  
  8e786f05
- 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