ml_switcheroo.compiler.frontends.rdna.nodes

RDNA AST Nodes.

Defines the data structures for the AMD RDNA / GCN syntax tree. Each node corresponds to a syntactic element in RDNA assembly and implements __str__ to emit valid assembly code.

Classes

RdnaNode	Abstract base class for all RDNA AST nodes.
Operand	Base class for instruction operands (Registers, Immediates, etc.).
LabelRef	Represents a reference to a label (e.g. as a jump target) or generic identifier operand.
SGPR	Represents a Scalar General Purpose Register (e.g., s0, s10).
VGPR	Represents a Vector General Purpose Register (e.g., v0, v255).
Immediate	Represents a literal constant value.
Modifier	Represents an instruction modifier or attribute.
Memory	Represents a memory address operand, typically used in generic loads/stores.
Instruction	Represents a single RDNA operation line.
Label	Represents a jump target label.
Directive	Represents an assembler directive.
Comment	Represents a line comment.

Functions

c_SGPR(→ SGPR)	Helper to create a single SGPR.
c_VGPR(→ VGPR)	Helper to create a single VGPR.

Module Contents

class ml_switcheroo.compiler.frontends.rdna.nodes.RdnaNode

Bases: abc.ABC

Abstract base class for all RDNA AST nodes.

abstractmethod __str__() → str

Returns the valid RDNA string representation of the node.

class ml_switcheroo.compiler.frontends.rdna.nodes.Operand

Bases: RdnaNode

Base class for instruction operands (Registers, Immediates, etc.).

__str__() → str

Returns the valid RDNA string representation of the node.

class ml_switcheroo.compiler.frontends.rdna.nodes.LabelRef

Bases: Operand

Represents a reference to a label (e.g. as a jump target) or generic identifier operand.

name

The label identifier.

Type:

str

name: str

__str__() → str

Returns the valid RDNA string representation of the node.

class ml_switcheroo.compiler.frontends.rdna.nodes.SGPR

Bases: Operand

Represents a Scalar General Purpose Register (e.g., s0, s10).

index

The register index.

Type:

int

count

If > 1, represents a range/multi-register (e.g., s[0:3]).

Type:

int

index: int

count: int = 1

__str__() → str

Returns the valid RDNA string representation of the node.

class ml_switcheroo.compiler.frontends.rdna.nodes.VGPR

Bases: Operand

Represents a Vector General Purpose Register (e.g., v0, v255).

index

The register index.

Type:

int

count

If > 1, represents a range/multi-register (e.g., v[0:3]).

Type:

int

index: int

count: int = 1

__str__() → str

Returns the valid RDNA string representation of the node.

ml_switcheroo.compiler.frontends.rdna.nodes.c_SGPR(idx: int) → SGPR

Helper to create a single SGPR.

ml_switcheroo.compiler.frontends.rdna.nodes.c_VGPR(idx: int) → VGPR

Helper to create a single VGPR.

class ml_switcheroo.compiler.frontends.rdna.nodes.Immediate

Bases: Operand

Represents a literal constant value.

value

The numeric value.

Type:

Union[int, float]

is_hex

If True, renders as hex string (e.g., 0xff).

Type:

bool

value: int | float

is_hex: bool = False

__str__() → str

Returns the valid RDNA string representation of the node.

class ml_switcheroo.compiler.frontends.rdna.nodes.Modifier

Bases: Operand

Represents an instruction modifier or attribute. e.g., glc, slc, off.

name

The modifier string.

Type:

str

name: str

__str__() → str

Returns the valid RDNA string representation of the node.

class ml_switcheroo.compiler.frontends.rdna.nodes.Memory

Bases: Operand

Represents a memory address operand, typically used in generic loads/stores. RDNA often passes the address as a register or register pair, but explicit offset syntax exists.

Format: method[base + offset] or simply register references depending on op. This node models the bracketed syntax global_load_dword v0, v[1:2], off. Usually, RDNA just uses registers as operands, but sometimes specific addressing notation is used. Here we treat it as an explicit container if needed, primarily for offsets.

base

The base register.

Type:

Union[SGPR, VGPR]

offset

Immediate byte offset.

Type:

Optional[int]

base: SGPR | VGPR

offset: int | None = None

__str__() → str

Returns the valid RDNA string representation of the node.

class ml_switcheroo.compiler.frontends.rdna.nodes.Instruction

Bases: RdnaNode

Represents a single RDNA operation line.

Format: {opcode} {operands} Modifiers are treated as operands in the list for flexible placement.

opcode

The instruction mnemonic (e.g., “v_add_f32”, “s_mov_b32”).

Type:

str

operands

List of operand nodes.

Type:

List[Operand]

opcode: str

operands: List[Operand] = []

__str__() → str

Returns the valid RDNA string representation of the node.

class ml_switcheroo.compiler.frontends.rdna.nodes.Label

Bases: RdnaNode

Represents a jump target label.

Format: {name}:

name

The label identifier.

Type:

str

name: str

__str__() → str

Returns the valid RDNA string representation of the node.

class ml_switcheroo.compiler.frontends.rdna.nodes.Directive

Bases: RdnaNode

Represents an assembler directive.

Format: .{name} {params}

name

The directive name (e.g., “global_base”).

Type:

str

params

List of string parameters.

Type:

List[str]

name: str

params: List[str] = []

__str__() → str

Returns the valid RDNA string representation of the node.

class ml_switcheroo.compiler.frontends.rdna.nodes.Comment

Bases: RdnaNode

Represents a line comment. RDNA assembly uses ; for comments.

Format: ; {text}

text

The comment content.

Type:

str

text: str

__str__() → str

Returns the valid RDNA string representation of the node.