Skip to main content
Log in

Mojo module

tcgen05

This module includes utilities for working with the tensorcore 5th generation (tcgen05) instructions.

Aliases

check_blackwell_constraint

alias 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

Was this page helpful?