[BUG] CuTe DSL doesn't support with_shape? #2722
Description
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.