ml_switcheroo.compiler.frontends.rdna¶

RDNA Frontend Package.

Contains the parser and lifter logic for converting AMD RDNA/GCN assembly text into Abstract Syntax Trees (AST) and then into the Logical Graph IR.

Submodules¶

Classes¶

`Comment`	Represents a line comment.
`Directive`	Represents an assembler directive.
`Immediate`	Represents a literal constant value.
`Instruction`	Represents a single RDNA operation line.
`Label`	Represents a jump target label.
`LabelRef`	Represents a reference to a label (e.g. as a jump target) or generic identifier operand.
`Memory`	Represents a memory address operand, typically used in generic loads/stores.
`Modifier`	Represents an instruction modifier or attribute.
`Operand`	Base class for instruction operands (Registers, Immediates, etc.).
`RdnaNode`	Abstract base class for all RDNA AST nodes.
`SGPR`	Represents a Scalar General Purpose Register (e.g., s0, s10).
`VGPR`	Represents a Vector General Purpose Register (e.g., v0, v255).
`RdnaParser`	Recursive descent parser for AMD RDNA / GCN assembly.
`RdnaLexer`	Regex-based Lexer for AMD RDNA / GCN assembly.
`Token`	Represents a lexical unit.
`TokenType`	Enumeration of valid RDNA token types.
`RdnaLifter`	Reconstructs a LogicalGraph from a sequence of RDNA AST nodes.
`RdnaAnalyzer`	Analyzes sequences of RDNA instructions to reverse-engineer high-level parameters.

Functions¶

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

Package Contents¶

class ml_switcheroo.compiler.frontends.rdna.Comment[source]¶

Bases: RdnaNode

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

Format: ; {text}

text¶

The comment content.

Type:: str

text: str¶

__str__() → str[source]¶: Returns the valid RDNA string representation of the node.

class ml_switcheroo.compiler.frontends.rdna.Directive[source]¶

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[source]¶: Returns the valid RDNA string representation of the node.

class ml_switcheroo.compiler.frontends.rdna.Immediate[source]¶

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[source]¶: Returns the valid RDNA string representation of the node.

class ml_switcheroo.compiler.frontends.rdna.Instruction[source]¶

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[source]¶: Returns the valid RDNA string representation of the node.

class ml_switcheroo.compiler.frontends.rdna.Label[source]¶

Bases: RdnaNode

Represents a jump target label.

Format: {name}:

name¶

The label identifier.

Type:: str

name: str¶

__str__() → str[source]¶: Returns the valid RDNA string representation of the node.

class ml_switcheroo.compiler.frontends.rdna.LabelRef[source]¶

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[source]¶: Returns the valid RDNA string representation of the node.

class ml_switcheroo.compiler.frontends.rdna.Memory[source]¶

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[source]¶: Returns the valid RDNA string representation of the node.

class ml_switcheroo.compiler.frontends.rdna.Modifier[source]¶

Bases: Operand

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

name¶

The modifier string.

Type:: str

name: str¶

__str__() → str[source]¶: Returns the valid RDNA string representation of the node.

class ml_switcheroo.compiler.frontends.rdna.Operand[source]¶

Bases: RdnaNode

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

__str__() → str[source]¶: Returns the valid RDNA string representation of the node.

class ml_switcheroo.compiler.frontends.rdna.RdnaNode[source]¶

Bases: abc.ABC

Abstract base class for all RDNA AST nodes.

abstractmethod __str__() → str[source]¶: Returns the valid RDNA string representation of the node.

class ml_switcheroo.compiler.frontends.rdna.SGPR[source]¶

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[source]¶: Returns the valid RDNA string representation of the node.

class ml_switcheroo.compiler.frontends.rdna.VGPR[source]¶

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[source]¶: Returns the valid RDNA string representation of the node.

ml_switcheroo.compiler.frontends.rdna.c_SGPR(idx: int) → SGPR[source]¶: Helper to create a single SGPR.

ml_switcheroo.compiler.frontends.rdna.c_VGPR(idx: int) → VGPR[source]¶: Helper to create a single VGPR.

class ml_switcheroo.compiler.frontends.rdna.RdnaParser(code: str)[source]¶

Recursive descent parser for AMD RDNA / GCN assembly.

lexer¶

tokens¶

pos = 0¶

parse() → List[ml_switcheroo.compiler.frontends.rdna.nodes.RdnaNode][source]¶: Parses the entire code block.

class ml_switcheroo.compiler.frontends.rdna.RdnaLexer[source]¶

Regex-based Lexer for AMD RDNA / GCN assembly.

PATTERNS: List[Tuple[TokenType, str]]¶

regex_pairs¶

tokenize(text: str) → Generator[Token, None, None][source]¶

Tokenizes the input string.

Parameters:: text (str) – Raw RDNA source code.
Yields:: Token – Token objects.
Raises:: ValueError – If an unrecognized character sequence is encountered.

class ml_switcheroo.compiler.frontends.rdna.Token[source]¶

Represents a lexical unit.

kind¶

The type of token.

Type:: TokenType

value¶

The raw string value.

Type:: str

line¶

Line number in source (1-based).

Type:: int

column¶

Column number in source (1-based).

Type:: int

kind: TokenType¶

value: str¶

line: int¶

column: int¶

class ml_switcheroo.compiler.frontends.rdna.TokenType(*args, **kwds)[source]¶

Bases: enum.Enum

Enumeration of valid RDNA token types.

LABEL_DEF¶

DIRECTIVE¶

COMMENT¶

COMMA¶

RBRACKET¶

LBRACKET¶

COLON¶

SGPR¶

VGPR¶

SPECIAL_REG¶

IMMEDIATE¶

MODIFIER¶

IDENTIFIER¶

class ml_switcheroo.compiler.frontends.rdna.RdnaLifter[source]¶

Reconstructs a LogicalGraph from a sequence of RDNA AST nodes.

lift(nodes: List[ml_switcheroo.compiler.frontends.rdna.nodes.RdnaNode]) → ml_switcheroo.compiler.ir.LogicalGraph[source]¶: Parses a list of RDNA nodes to build a LogicalGraph.

class ml_switcheroo.compiler.frontends.rdna.RdnaAnalyzer[source]¶

Analyzes sequences of RDNA instructions to reverse-engineer high-level parameters.

static analyze_block(kind: str, instructions: List[ml_switcheroo.compiler.frontends.rdna.nodes.Instruction]) → Dict[str, Any][source]¶

Extracts metadata from a block of instructions based on the operation kind.

Parameters:

kind (str) – The operation type (e.g. “Conv2d”, “Linear”).
instructions (List[Instruction]) – The assembly lines inside the block.

Returns:

Extracted parameters (e.g., {“k”: 3}).

Return type:

Dict[str, Any]