Lightning-AI/lightning-thunder

Handle returning NamedTuples from the JIT

开放

#147 创建于 2024年4月9日

 (0 条评论) (0 个反应) (0 位负责人)Python (114 个派生)github user discovery
enhancementhelp wantedjit

仓库指标

星标
 (1,460 个星标)
PR 合并指标
 (30 天内没有已合并 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.

贡献者指南