Skip to content

Commit faa176f

Browse files
committed
More barrier functionality
1 parent 0079d3a commit faa176f

File tree

1 file changed

+14
-2
lines changed

1 file changed

+14
-2
lines changed

lib/mtl/command_enc4.jl

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,21 @@ export endEncoding!, barrierAfterStages!
22

33
# @objcwrapper immutable=true MTL4CommandEncoder <: NSObject
44

5-
function barrierAfterStages!(encoder::MTL4CommandEncoder, afterStages::MTLStages=MTLStageAll, queueStages::MTLStages=MTLStageAll, visibilityOptions::MTL4VisibilityOptions=MTL4VisibilityOptionDevice)
5+
function barrierAfterEncoderStages!(encoder::MTL4CommandEncoder, afterEncoderStages::MTLStages=MTLStageAll, beforeEncoderStages::MTLStages=MTLStageAll, visibilityOptions::MTL4VisibilityOptions=MTL4VisibilityOptionResourceAlias)
6+
@objc [encoder::id{MTL4CommandEncoder} barrierAfterEncoderStages:afterEncoderStages::MTLStages
7+
beforeEncoderStages:beforeEncoderStages::MTLStages
8+
visibilityOptions:visibilityOptions::MTL4VisibilityOptions]::Nothing
9+
end
10+
11+
function barrierAfterQueueStages!(encoder::MTL4CommandEncoder, afterQueueStages::MTLStages=MTLStageAll, beforeStages::MTLStages=MTLStageAll, visibilityOptions::MTL4VisibilityOptions=MTL4VisibilityOptionResourceAlias)
12+
@objc [encoder::id{MTL4CommandEncoder} barrierAfterQueueStages:afterQueueStages::MTLStages
13+
beforeStages:beforeStages::MTLStages
14+
visibilityOptions:visibilityOptions::MTL4VisibilityOptions]::Nothing
15+
end
16+
17+
function barrierAfterStages!(encoder::MTL4CommandEncoder, afterStages::MTLStages=MTLStageAll, beforeQueueStages::MTLStages=MTLStageAll, visibilityOptions::MTL4VisibilityOptions=MTL4VisibilityOptionResourceAlias)
618
@objc [encoder::id{MTL4CommandEncoder} barrierAfterStages:afterStages::MTLStages
7-
beforeQueueStages:queueStages::MTLStages
19+
beforeQueueStages:beforeQueueStages::MTLStages
820
visibilityOptions:visibilityOptions::MTL4VisibilityOptions]::Nothing
921
end
1022

0 commit comments

Comments
 (0)