[FEA] Tensor manipulation methods for `cute.Tensor`
Nobody has claimed this yet.
- Dominant language
- C++
- Stars
- 10.5k
- Forks
- 2.1k
- Avg merge
- 3d 11h
- Merged PRs (30d)
- 7
Description
Which component requires the feature?
CuTe DSL
Feature Request
Is your feature request related to a problem? Please describe.
It would be useful for cute.Tensor to support permute, transpose, unsqueeze, expand. Calling them in host-side launchers instead of torch would reduce CPU-side overheads.
Describe the solution you'd like
Expand the cute.Tensor class to include the methods suggested.
Describe alternatives you've considered
Defining helper methods
Additional context
N/A
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start with the CuTe DSL's cute.Tensor class and inspect how host-side launchers currently use torch for tensor manipulation. Determine the expected behavior of permute, transpose, unsqueeze, and expand, then verify that the methods support the requested launcher use cases without relying on helper methods.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Quiet
- Clarity
- Mostly clear
- Newbie friendliness
- 48/100