[BUG] CuTe DSL doesn't support with_shape?
Which component has the problem?
CuTe DSL
Bug Report
Describe the bug
Steps/Code to reproduce bug
import cutlass.cute as cute
@cute.jit
def test():
layoutA = cute.make_layout((4, 4), stride=(4, 1))
layoutA_inv = cute.left_inverse(layoutA)
layoutA_inv_with_shape = layoutA_inv.with_shape((8, 2))
print(layoutA)
print(layoutA_inv)
print(layoutA_inv_with_shape)
test()
Expected behavior A clear and concise description of what you expected to happen.
Environment details (please complete the following information):
- Environment location: [Bare-metal, Docker, Cloud(specify cloud provider)]
Additional context Add any other context about the problem here.
Not yet. Can be supported quickly. Before that I think you can use
cute.composition(layoutA_inv, cute.make_layout((8, 2)))
@brandon-yujie-sun for visibility
This issue has been labeled inactive-30d due to no recent activity in the past 30 days. Please close this issue if no further response or action is needed. Otherwise, please respond with a comment indicating any updates or changes to the original issue and/or confirm this issue still needs to be addressed. This issue will be labeled inactive-90d if there is no activity in the next 60 days.