[QST] [CuTeDSL] Object aliasing breaks inside `cutlass.range`

[HanGuo97]

What is your question?

Hi, I noticed that when using dynamic range, object aliasing seems to break if inplace operations are performed. Is this expected?

Sample code:

import cutlass
import cutlass.cute as cute

@cute.jit
def f():
    state0 = cutlass.pipeline.make_pipeline_state(cutlass.pipeline.PipelineUserType.Producer, stages=2)
    state1 = state0
    print(state1 is state0)
    state0.advance()
    print(state1 is state0)  # -> True
    state0.advance()
    print(state1 is state0)  # -> True
    state0.advance()
    print(state1 is state0)  # -> True

    for i in cutlass.range_constexpr(3):
        state0.advance()
        print(state1 is state0)  # --> True
    print(state1 is state0)  # --> True

    print("--- Here! ---")
    for i in cutlass.range(3):
        state0.advance()
        print(state1 is state0)  # --> False!
    print(state1 is state0)  # --> False!
f()

[anakinxc]

Hi @HanGuo97

Thanks for reporting this, it should be treated as a bug.

[HanGuo97]

got it, this is to to know, so I will keep this open.

Do you know if this is specific to pipeline or inplace operations on any objects?

[anakinxc]

This is related to how objects are handled across dynamic control flows

[HanGuo97]

ok, thanks again. Will find some workarounds for now!