Skip to content

Extend tracks functionality for Composite Ops #1354

@ricardoV94

Description

@ricardoV94
Member

We probably can hack the tracks functionality for these sort of composite ops, we could do something like BlockwiseOf(CoreOp) dummy type and have special logic in the tracks?

Originally posted by @ricardoV94 in #1353 (comment)

This opens the door to the idea of making rewrites more "database index-like". Stuff like SequenceOf(Solve, Diagonal), that the PatternNodeRewriter tries to do, but fails for the very same kind of parametrized Ops (which solve type should it match?).

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

        @ricardoV94

        Issue actions

          Extend tracks functionality for Composite Ops · Issue #1354 · pymc-devs/pytensor