Skip to content

Cache output of numba_funcify where possible #180

Open
@aseyboldt

Description

@aseyboldt
Member

We could add cache_by_op or cache_by_op_and_sig decorators for numba_funcitfy functions, that internally use a LRU cache to keep track of previous njit functions. Each funcify function can then choose to reuse results in a way that is appropriate.
Most funcify functions only use the op, and don't even look at the node at all.
See also #179 for context.

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

      Development

      No branches or pull requests

        Participants

        @aseyboldt@ricardoV94

        Issue actions

          Cache output of numba_funcify where possible · Issue #180 · pymc-devs/pytensor