Mojo module
runtime_tuple
Provides the RuntimeTuple data structure and related utility functions for handling tuple-like data with both compile-time and runtime elements. RuntimeTuple is designed for high-performance tensor operations, supporting efficient manipulation of multi-dimensional data structures like shapes, indices, and coordinates.
Key features:
- Hybrid compile-time/runtime value handling
- Optimized for parallel execution and hardware acceleration
- Support for nested tuple structures
- Efficient conversion between linear indices and multi-dimensional coordinates
- Specialized operations for tensor shape calculations
The module includes functions for tuple manipulation (concatenation, flattening),
coordinate transformations (idx2crd, crd2idx), and specialized tensor operations
like shape division and prefix products.
Structs
-
RuntimeTuple: A struct representing tuple-like data with compile-time and runtime elements.
Functions
-
concat: Concatenates twoIntTupleinstances into a singleIntTuple. -
crd2idx: Converts multi-dimensional coordinates to a linear index. -
idx2crd: Converts a linear index to multi-dimensional coordinates. -
is_int: Determines if aRuntimeTuplerepresents a scalar integer value. -
is_tuple: Determines if aRuntimeTuplerepresents a tuple rather than a scalar value. -
prefix_product: Computes the prefix products of elements in theRuntimeTuple. -
product: Computes the product of all elements in theRuntimeTuple. -
shape_div: Performs specialized shape division betweenRuntimeTuples. -
signum: Returns the sign of an integer value.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!