Skip to main content

Mojo module

layout_tensor

Provides the LayoutTensor type for representing multidimensional data.

Aliases

binary_op_type

alias binary_op_type = fn[DType, Int](lhs: SIMD[$0, $1], rhs: SIMD[$0, $1]) -> SIMD[$0, $1]

Type alias for binary operations on SIMD vectors.

This type represents a function that takes two SIMD vectors of the same type and width and returns a SIMD vector of the same type and width.

Args: dtype: The data type of the SIMD vector elements. width: The width of the SIMD vector. lhs: Left-hand side SIMD vector operand. rhs: Right-hand side SIMD vector operand.

Returns: A SIMD vector containing the result of the binary operation.

Structs

  • LayoutTensor: A high-performance tensor with explicit memory layout and hardware-optimized access patterns.
  • LayoutTensorIter: Iterator for traversing a memory buffer with a specific layout.
  • ThreadScope: Represents the scope of thread operations in GPU programming.

Functions

Was this page helpful?