burn icon indicating copy to clipboard operation
burn copied to clipboard

Published 20 hours ago • verified publisher icon tracel-ai

Add cumsum tensor op

Open laggui opened this issue 11 months ago • 1 comments

Checklist

  • [ ] Confirmed that run-checks all script has been executed.
  • [x] Made sure the book is up to date with changes in this PR.

Related Issues/PRs

#1722 #2099

Changes

Added tensor.cumsum(dim) operation

TODO:

  • [ ] JIT implementation w/ cubecl kernel
    • We have cubecl-reduce at this time but we might need to add cubecl-scan for this family of ops (some refs: wgsl simd cuda)

Testing

Unit test.

laggui avatar Jan 06 '25 17:01 laggui

This PR has been marked as stale because it has not been updated for over a month

github-actions[bot] avatar Apr 03 '25 12:04 github-actions[bot]

This op is very important for many NAR TTS models, and it is blocking porting one of my open-source TTS models to burn.

mush42 avatar May 30 '25 21:05 mush42

This PR has been marked as stale because it has not been updated for over a month

github-actions[bot] avatar Jun 30 '25 12:06 github-actions[bot]

This PR has been marked as stale because it has not been updated for over a month

github-actions[bot] avatar Sep 03 '25 12:09 github-actions[bot]

So much has changed, we should probably create a new PR by now.

antimora avatar Sep 24 '25 15:09 antimora