iron.kernels.linalg Namespace Reference

Variables
dict	_CASCADE_COMBOS
dict	_CASCADE_MM_SCALAR_DIMS

Detailed Description

Linear algebra kernel factories: mm, mv, cascade_mm.

Variable Documentation

_CASCADE_COMBOS

dict iron.kernels.linalg._CASCADE_COMBOS

protected

Initial value:

=  {
    (np.int16, np.int16): "i16_i16",
    (np.int16, np.int32): "i16_i32",
    (bfloat16, bfloat16): "bf16_bf16",
    (bfloat16, np.float32): "bf16_f32",
}

_CASCADE_MM_SCALAR_DIMS

dict iron.kernels.linalg._CASCADE_MM_SCALAR_DIMS

protected

Initial value:

=  {
    (np.int16, np.int16): (1, 1, 1),
    (np.int16, np.int32): (1, 1, 1),
    (bfloat16, bfloat16): (1, 1, 1),
    (bfloat16, np.float32): (1, 1, 1),
}