EnzymeAD / EnzymeAD/Reactant.jl

Speed up traced_type by conversion to generated function

Open
#2,057 2 comments 0 reactions 0 assignees View on GitHub
Dominant language
Julia
Stars
370
Forks
74
Avg merge
18h 47m
Merged PRs (30d)
30

Description

Can be done analagously to Enzyme.active_reg_inner. This means we will have type inference return a constant type here (and make traced_type and the rest much faster/infer). Moreover we can leverage a global cache for all type queries. It will be invalidated upon a new def of traced_type_inner.

This should speed up all calls to traced_type.

It does require traced_type to be computable as a function of type arguments.

@avik-pal do we need the sharded value for traced_type_inner or does the sharded type suffice?

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.