For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).
Mojo struct
TensorMemory
struct TensorMemory
A wrapper around tensor memory allocated for tcgen05 instructions.
Fields
- ptr (
Pointer[UInt32, MutUntrackedOrigin, address_space=AddressSpace.SHARED, _safe=True]): Pointer to the tensor memory address. - num_cols (
UInt32): The number of columns in the tensor memory.
Implemented traits
AnyType,
Copyable,
ImplicitlyCopyable,
ImplicitlyDeletable,
Movable,
RegisterPassable,
TrivialRegisterPassable
Methods
__init__
def __init__(num_cols: UInt32) -> Self
Initialize the TensorMemory struct.
Args:
- num_cols (
UInt32): The number of columns to allocate.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!