Lightning-AI / Lightning-AI/lightning-thunder

Connect `prims.copy_with_setitem` to nvFuser's Executor

Open
#1,676 1 comment 0 reactions 1 assignee Claimed by @kevinstephano View on GitHub
enhancement nvfuser thunderfx
Dominant language
Python
Stars
1.5k
Forks
121
PR merge metrics
No merged PRs in 30d

Description

## 🚀 Feature
https://github.com/Lightning-AI/lightning-thunder/blob/bed8e731e958d693244cb303f306ee1f469fed6f/thunder/core/prims.py#L3465-L3470
It would be good to connect `prims.copy_with_setitem` to nvFuser's executor. How is setitem supposed to be implemented?

### Motivation

Would help reduce the exposure of Thunder overheads for the execution of individual operations.

cc @tfogal

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.