lightning-thunder icon indicating copy to clipboard operation
lightning-thunder copied to clipboard
verified publisher icon Lightning-AI

Support CUDA stream operators in ThunderFX

Open kiya00 opened this issue 1 month ago • 2 comments

Before submitting
  • [ ] Was this discussed/approved via a Github issue? (no need for typos and docs improvements)
  • [ ] Did you read the contributor guideline, Pull Request section?
  • [ ] Did you make sure to update the docs?
  • [ ] Did you write any new necessary tests?

What does this PR do?

Fixes #2332 .

Support CUDA stream operators in ThunderFX This PR added the patch mentioned in #2332 and a follow-up fix in https://github.com/Lightning-AI/lightning-thunder/pull/2750#discussion_r2538168812

kiya00 avatar Nov 20 '25 13:11 kiya00

Is there a small repro that we can use to test this PR?

Yeah, I add a case to test it

kiya00 avatar Nov 21 '25 16:11 kiya00

Hi @KaelanDt , could you help review this?

kiya00 avatar Nov 25 '25 09:11 kiya00

Hi @KaelanDt , I think it's ready to merge, could you help to take a look?

kiya00 avatar Nov 27 '25 10:11 kiya00