Mojo module
tcgen05
This module includes utilities for working with the tensorcore 5th generation (tcgen05) instructions.
Aliases
-
check_blackwell_constraint = constrained[::Bool,::StringSlice[::Bool[_has_blackwell_tcgen05(), __init__[__mlir_type.!kgen.string]("The tcgen05 instructions are only applicable on nVidia Blackwell (sm_100a, sm_101a) hardware."), ?]:
Structs
-
TensorMemory: A wrapper around tensor memory allocated for tcgen05 instructions.
Functions
-
tcgen05_alloc: Allocates tensor memory for use with tcgen05 instructions. -
tcgen05_dealloc: Deallocates tensor memory allocated by tcgen05_alloc().
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!