Lightning-AI/lightning-thunder

Handle returning NamedTuples from the JIT

オープン

#147 opened on 2024/04/09

 (0 件のコメント) (0 件のリアクション) (0 人の担当者)Python (114 件のフォーク)github user discovery
enhancementhelp wantedjit

Repository metrics

Stars
 (1,460 個のスター)
PR merge metrics
 (30d に merged PR はありません)

説明

NamedTuples have (some) support in the interpreter, but to return them from the JIT, we would need some more things:

  • ensure value tracking covers creation (populating the attribute_wrappers) - this is in interpreter,
  • when seeing NamedTuples being returned, add epilogue code to create the NamedTuple from its contents (which are available),

This is nontrivial as it requires return values to be routed through the epilogue, but I don't think that is grave.

This is probably an intermediate to advanced issue for people fond of the JIT / frontend bits.

コントリビューターガイド