diff --git a/lib/mtl/buffer.jl b/lib/mtl/buffer.jl index a46fb220b..aad8fd14d 100644 --- a/lib/mtl/buffer.jl +++ b/lib/mtl/buffer.jl @@ -1,4 +1,4 @@ -export MTLBuffer, device, handle +export MTLBuffer # From docs: "MSL implements a buffer as a pointer to a built-in or user defined data type described in the # device, constant, or threadgroup address space. diff --git a/lib/mtl/command_enc/compute.jl b/lib/mtl/command_enc/compute.jl index 39a3deb0c..ee91c97d4 100644 --- a/lib/mtl/command_enc/compute.jl +++ b/lib/mtl/command_enc/compute.jl @@ -1,6 +1,6 @@ export MTLComputeCommandEncoder -export set_function!, set_buffer!, set_bytes!, dispatchThreads!, endEncoding! -export set_buffers!, append_current_function! +export set_function!, set_buffer!, dispatchThreadgroups!, endEncoding! +export append_current_function! @cenum MTLDispatchType::NSUInteger begin MTLDispatchTypeSerial = 0 diff --git a/src/array.jl b/src/array.jl index 741482585..c938e6599 100644 --- a/src/array.jl +++ b/src/array.jl @@ -1,6 +1,6 @@ # host array -export MtlArray, MtlVector, MtlMatrix, MtlVecOrMat, mtl, is_shared, is_managed, is_private +export MtlArray, MtlVector, MtlMatrix, MtlVecOrMat, mtl, is_shared, is_managed, is_private, device function hasfieldcount(@nospecialize(dt)) try