Mojo module

apple_amx_intrinsics

Functions

dot_at_b: Performs a matrix multiply C = A^T * B using Apple AMX instructions.
extrx: Extracts a row or moves it to x, result in amx0.
extry: Extracts a row or moves it to y, result in amx0.
fma:
fma16: Float16 matrix multiply and subtract.
fma32: Float32 matrix multiply and add.
fma64: Float64 matrix multiply and add.
fms16: Float16 matrix multiply and add.
fsm32: Float32 matrix multiply and subtract.
fsm64: Float64 matrix multiply and subtract.
genlut:
ldx:
ldy:
ldz:
ldzi:
load_z:
mac16: SI16 matrix multiply and add.
matfp: Float16 matrix multiply.
max_int__: UI16 matrix multiply.
read_x:
read_y:
store_x:
store_y:
store_z:
stx:
sty:
stz:
stzi:
transpose_z_to_x_or_y:
vec_int__: Horizontal ui16 multiply z0[i] += x0[i] + y0[i].
vecfp: Horizontal float16 multiply z0[i] += x0[i] + y0[i].