struct

ML::Metal::ComputeGraph::OpBuilder

Inherits Struct / Value / Object

Builder passed to the add_op block for fluent op construction

Instance methods

buffer(buf : MetalBuffer, index : Int32, access : BufferAccess = BufferAccess::Read, offset : Int64 = 0) : OpBuilder
Source
dispatch_1d(count : Int32, tg_size : Int32 = 256) : Nil
Source
dispatch_indirect(buf : MetalBuffer, offset : Int64, size : Tuple(Int32, Int32, Int32)) : Nil
Source
dispatch_threadgroups(grid : Tuple(Int32, Int32, Int32), size : Tuple(Int32, Int32, Int32)) : Nil
Source
threadgroup_memory(bytes : Int32) : OpBuilder
Source
value(val : UInt32, index : Int32) : OpBuilder
Source
value(val : Float32, index : Int32) : OpBuilder
Source