Lightning-AI/lightning-thunder

Error on reentrant jit invocation

Open

#1,222 opened on Sep 30, 2024

 (7 comments) (0 reactions) (0 assignees)Python (114 forks)github user discovery
enhancementgood first issueux

Repository metrics

Stars
 (1,460 stars)
PR merge metrics
 (No merged PRs in 30d)

Description

...to provide a better error message. Most likely the easiest is to add something like

    if get_jit_ctx() is not None:
        raise NotImplementedError("Re-entrant jitting is not supported")

to the top of thunder.jit.thunder_general_jit and then write a test.

First step in #1126

cc @carmocca @borda

Contributor guide