arxjit.types

arxjit.types

Source: packages/arxjit/src/arxjit/types.py

title: Scalar type and signature API for arxjit.
summary: >-
  Defines the built-in scalar types (``i64``, ``f64`` ...) that users reference
  in ``@jit`` signatures, and the ``Signature`` produced by calling a return
  type with argument types, e.g. ``i64(i64, i64)``.

Classes

SigType

title: A type usable in an arxjit signature.
summary: >-
  Scalar today (e.g. ``i64``); the name is kept element-vs-array neutral so
  array forms such as ``i64[2, 2]`` can be added later via subscripting
  without renaming the public type.
attributes:
  name:
    type: str
    description: The arxjit-facing name, e.g. ``i64``.
  astx_name:
    type: str
    description: The astx class this type lowers to, e.g. ``Int64``.

Signature

title: A compiled-function signature (return type and argument types).
attributes:
  return_type:
    type: SigType
    description: The type returned by the compiled function.
  arg_types:
    type: tuple[SigType, Ellipsis]
    description: The positional argument types, in order.