MatrixDimInfo
public struct MatrixDimInfo
This struct corresponds to the MatrixDimInfo
struct in file ‘matrix_mult_op.metal’
-
Undocumented
Declaration
Swift
var M: MetalUInt // number of rows in A
-
Undocumented
Declaration
Swift
var N: MetalUInt // number of cols in B
-
Undocumented
Declaration
Swift
var K: MetalUInt // number of cols in A, number of rows in B
-
Undocumented
Declaration
Swift
var stride: MetalUShort // element stride in bytes
-
Undocumented
Declaration
Swift
var beta: MetalFloat // scale of output