Infer type for builtin functions with eval_return_type.

Infer C++ type from Python value node.

Infer type for LLKernel calls.

Infer type for method calls (e.g., obj.method()).

Infer type of a return expression node.

Infer type of a special struct attribute by analyzing __init__ method.

Infer type for special struct constructors.

Handle unresolved function calls (method calls on variables).

Initialize CUDA context.

Check if method should be const (doesn't modify member variables).

Check if condition should use constexpr if (template parameters, enum comparisons, or ll.const).

True if value is safe to add to ctx (whitelist for closure vars).

Check if a variable name is defined locally (should not be resolved from ctx). Supports both: - scope_vars (Dict[str

Check if a parameter is modified in the method body.

Load libcuda.so library with robust path detection. Tries multiple methods to find libcuda: 1. Environment variable LITTLE_KE

Load libcuda.so library.

rand() * scale + bias randint(-scale, scale) + bias

Map call arguments (positional/keyword) to function parameters

$ rocm-smi --shownodesbw ============================ ROCm System Management Interface ============================ ===============

Triton Dist PP get operation based on test_pp.py

Triton Dist PP put operation based on test_pp.py

Convert a Python default value to an AST node. Handles: - Basic types (int, float, bool, str, None) -> ast.Constant - Enum insta

qkv: (bs, seq_len, num_total_heads, head_dim) BLOCK_HD equal to next_power_of_2(head_dim)

split src/dest outside of _recv_ll. this function is designed for a threadblock num_ints: of the pre-LL-packed num_ints.

Register default codegen handlers for loop modifiers.

Register default loop modifiers.

Register default operators that return the left operand type.

Replace ll.empty assigns with alloc/slice in body and nested blocks. No hardcoded sizes.

Replace return statements with assignments to the LHS variable (for functions with return values)

Rename local variables in inlined function to avoid conflicts with: - Variables in the calling context - Parameters of the ca

Resolve the function being called.

Resolve subscript annotations like template[...], Tuple[...], or const[...]

Generic function to resolve a template parameter from AST node to C++ constant expression. This is a generic version that works for any

Initializes buffers for communication.

Executes one fundamental step of the pipeline for the current stage.

Replace function parameters with call arguments and rename conflicting variables