Code

Cache mapped functions#531

Code

Merged

inducer merged 13 commits intoinducer:maininducer/pytato:mainfrom majosm:cache-mapped-function-resultsmajosm/pytato:cache-mapped-function-resultsCopy head branch name to clipboardJan 17, 2025

+334-198

Commits

Commits on Jan 9, 2025

• avoid traversing functions multiple times

  majosm

  committed
  eb75be0

  Copy full SHA for eb75be0

• condense function cache setup

  majosm

  committed
  786088c

  Copy full SHA for 786088c

• pass function cache to CombineMapper.__init__ too

  majosm

  committed
  2108afd

  Copy full SHA for 2108afd

• use Never instead of None as the function result type for mappers that don't support functions

  majosm

  committed
  1596db4

  Copy full SHA for 1596db4

• remove multiple inheritance for TransformMapperWithExtraArgs

  Show description for 2cac57f
  majosm

  committed
  2cac57f

  Copy full SHA for 2cac57f

• add _verify_is_array to avoid the need for rec_ary

  Show description for 40b47e9
  majosm

  committed
  40b47e9

  Copy full SHA for 40b47e9

• remove map_foreign from Mapper

  majosm

  committed
  1890af8

  Copy full SHA for 1890af8

Commits on Jan 13, 2025

• use Never as FunctionResultT for NumpyCodegenMapper and FancyDotWriter too

  majosm

  committed
  4e9a4d0

  Copy full SHA for 4e9a4d0

• make PlaceholderSubstitutor explicit about not supporting functions

  majosm

  committed
  d3671b4

  Copy full SHA for d3671b4

Commits on Jan 16, 2025

• tweak types in rec/rec_function_definition

  majosm

  committed
  a7e4797

  Copy full SHA for a7e4797

• use P.args/P.kwargs in CachedWalkMapper

  majosm

  committed
  0d18657

  Copy full SHA for 0d18657

• use a more specific type than Any for cache key in CachedWalkMapper

  majosm

  committed
  b704ede

  Copy full SHA for b704ede

• remove some more Anys

  majosm

  committed
  d912360

  Copy full SHA for d912360