Class Var

python/little_kernel/core/expr.py:246–261

Variable expression (access to local/global variables or parameters)

class Var(Expr):
    """Variable expression (access to local/global variables or parameters)"""

    def __init__(self, name: str, expr_type: LLType, defining_stmt=None, is_shared: bool = False,
                 is_global: bool = False):
        super().__init__(expr_type)
        self.name = name  # Variable name
        self.defining_stmt = defining_stmt
        self.is_shared = is_shared  # Whether it's a __shared__ variable
        self.is_global = is_global  # Whether it's a __device__ global variable

    def __repr__(self) -> str:
        return f"Var(name={self.name}, type={self.type}, shared={self.is_shared}, global={self.is_global})"

    def __str__(self) -> str:
        return f"Var(name={self.name}, type={self.type}, shared={self.is_shared}, global={self.is_global})"


# Unary