tvm_tirx.backend.cuda.lang.PipelineState

class tvm_tirx.backend.cuda.lang.PipelineState(*args, **kwargs)

Bases: object

Tracks stage and phase for a software-pipelined ring buffer.

This class does not know anything about full/empty barriers. Use it when the kernel manually waits/signals barriers, or when the stage/phase drives a ring not wrapped in a Pipeline.

Parameters:

• depth (int) – Number of stages in the ring.

• phase (int, optional) – Initial phase. Omit when initialization should happen later.

init(phase)

advance()