Commits · d654c56bf7fc2e68bab7c658e2de982e582f051f · Andreas Klöckner / pytato

Browse files

Jun 01, 2023
- Implement pt.inline_calls · d654c56b
  Kaushik Kulkarni authored Nov 05, 2022 and Andreas Klöckner committed Jun 01, 2023
  
  d654c56b
- Support mapper methods for FunctionDefintion, Call · b1f8dba2
  Kaushik Kulkarni authored Nov 04, 2022 and Andreas Klöckner committed Jun 01, 2023
  b1f8dba2
- Introduce `pt.trace_call` · 6254afca
  Kaushik Kulkarni authored Nov 04, 2022 and Andreas Klöckner committed Jun 01, 2023
  6254afca
May 25, 2023
- Install pytest for Conda Examples Githhub CI job · 0172cca8
  Andreas Klöckner authored May 25, 2023
  
  0172cca8
May 23, 2023
- test visualizers · a6240512
  Kaushik Kulkarni authored May 23, 2023 and Andreas Klöckner committed May 23, 2023
  a6240512
- Implement fancy placeholder focused data-flow viz · b953db03
  Kaushik Kulkarni authored May 23, 2023 and Andreas Klöckner committed May 23, 2023
  
  b953db03
- account for weak dependency on mako · 8240db16
  Kaushik Kulkarni authored May 23, 2023 and Andreas Klöckner committed May 23, 2023
  
  8240db16
- adds helper to obtain the subscript of an einsum · 1af435c1
  Kaushik Kulkarni authored May 23, 2023 and Andreas Klöckner committed May 23, 2023
  
  1af435c1
May 22, 2023
- split visualization.py · 8902f9fc
  Kaushik Kulkarni authored May 22, 2023 and Andreas Klöckner committed May 22, 2023
  8902f9fc
May 18, 2023
- test apply_distributive_property_to_einsums · ac05808f
  Kaushik Kulkarni authored May 09, 2023 and Andreas Klöckner committed May 17, 2023
  ac05808f
- Implement distributive law along an einsum's args · 2c443cef
  Kaushik Kulkarni authored May 09, 2023 and Andreas Klöckner committed May 17, 2023
  
  2c443cef
- rewrite matmul using einsum · 8b0eff23
  Kaushik Kulkarni authored May 09, 2023 and Andreas Klöckner committed May 17, 2023
  
  8b0eff23
May 11, 2023
- Force distributed example to run multi-rank · babbbb62
  Matt Smith authored Jan 18, 2023 and Andreas Klöckner committed May 11, 2023
  babbbb62
- add test for periodic communication · 004b56f1
  Matt Smith authored Feb 07, 2023 and Andreas Klöckner committed May 11, 2023
  
  004b56f1
- add test for materialized arrays promoted to part outputs · ae93d644
  Matt Smith authored Jan 31, 2023 and Andreas Klöckner committed May 11, 2023
  
  ae93d644
- add test for dag with multiple send nodes per sent array · 031f1e0b
  Matt Smith authored Jan 18, 2023 and Andreas Klöckner committed May 11, 2023
  
  031f1e0b
- add test for dag with a receive as an output · c602fbdb
  Matt Smith authored Jan 18, 2023 and Andreas Klöckner committed May 11, 2023
  
  c602fbdb
- add test for dag with duplicated output array · 94d1efbf
  Matt Smith authored Jan 18, 2023 and Andreas Klöckner committed May 11, 2023
  
  94d1efbf
- Add a test for Kaushik's distributed MWE · f6b83576
  Kaushik Kulkarni authored Jan 11, 2023 and Andreas Klöckner committed May 11, 2023
  
  f6b83576
- New distributed-memory DAG partitioner · 6636a0ff
  Andreas Klöckner authored May 03, 2023 and Andreas Klöckner committed May 11, 2023
  6636a0ff
- run_test_with_mpi: Allow passing extra_env_vars · 53b124e3
  Andreas Klöckner authored Jan 12, 2023 and Andreas Klöckner committed May 11, 2023
  53b124e3
Apr 25, 2023
- switch to new intersphinx mapping format · 0566d597
  Kaushik Kulkarni authored Apr 25, 2023 and Andreas Klöckner committed Apr 25, 2023
  
  0566d597
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