Skip to main content
Version: Nightly

TensorMemory

struct TensorMemory

A wrapper around tensor memory allocated for tcgen05 instructions.

Fields​

  • ​ptr (UnsafePointer[UInt32, MutAnyOrigin, address_space=AddressSpace.SHARED]): Pointer to the tensor memory address.
  • ​num_cols (UInt32): The number of columns in the tensor memory.

Implemented traits​

AnyType, Copyable, ImplicitlyCopyable, ImplicitlyDestructible, Movable, RegisterPassable, TrivialRegisterPassable

Methods​

__init__​

__init__(num_cols: UInt32) -> Self

Initialize the TensorMemory struct.

Args:

  • ​num_cols (UInt32): The number of columns to allocate.
Edit this page