More api

Author: christiangnrd

lib/mtl/command_enc/compute4.jl

@@ -1,5 +1,6 @@
 export MTL4ComputeCommandEncoder
 export set_function!, set_buffer!, set_bytes!, set_texture!, set_sampler_state!
+export stages
 export dispatchThreadgroups!, dispatchThreads!, endEncoding!
 export use!, memoryBarrier!, append_copy!, append_fillbuffer!, append_sync!
 
@@ -67,6 +68,10 @@ end
 #                                              destinationOrigin:dstOrigin::MTLOrigin]::Nothing
 # end
 
+function stages(cce::MTL4ComputeCommandEncoder)
+    @objc [cce::id{MTL4ComputeCommandEncoder} stages]::MTLStages
+end
+
 # Fill Buffer
 function append_fillbuffer!(cce::MTL4ComputeCommandEncoder, buffer::MTLBuffer, range::NSRange, value::UInt8)
     @objc [cce::id{MTL4ComputeCommandEncoder} fillBuffer:buffer::id{MTLBuffer}

lib/mtl/command_enc4.jl

@@ -1,7 +1,18 @@
-export endEncoding!, barrierAfterStages!
+export endEncoding!, updateFence!, waitForFence
+export barrierAfterEncoderStages!, barrierAfterQueueStages!, barrierAfterStages!
 
 # @objcwrapper immutable=true MTL4CommandEncoder <: NSObject
 
+function updateFence!(encoder::MTL4CommandEncoder, fence::MTLFence, afterEncoderStages::MTLStages=MTLStageAll)
+    @objc [encoder::id{MTL4CommandEncoder} updateFence:fence::id{MTLFence}
+                                            afterEncoderStages:afterEncoderStages::MTLStages]::Nothing
+end
+
+function waitForFence(encoder::MTL4CommandEncoder, fence::MTLFence, beforeEncoderStages::MTLStages=MTLStageAll)
+    @objc [encoder::id{MTL4CommandEncoder} waitForFence:fence::id{MTLFence}
+                                            beforeEncoderStages:afterEncoderStages::MTLStages]::Nothing
+end
+
 function barrierAfterEncoderStages!(encoder::MTL4CommandEncoder, afterEncoderStages::MTLStages=MTLStageAll, beforeEncoderStages::MTLStages=MTLStageAll, visibilityOptions::MTL4VisibilityOptions=MTL4VisibilityOptionResourceAlias)
     @objc [encoder::id{MTL4CommandEncoder} barrierAfterEncoderStages:afterEncoderStages::MTLStages
                                             beforeEncoderStages:beforeEncoderStages::MTLStages