We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 383102f commit e2368dcCopy full SHA for e2368dc
src/nn/vulkan/silu-forward-f32-f32.comp
@@ -15,20 +15,17 @@ layout(binding = 0) readonly buffer inputBuffer { float x[]; };
15
layout(binding = 1) writeonly buffer outputBuffer { float y[]; };
16
layout(binding = 2) readonly buffer batchInfosBuffer { BatchInfo infos[]; };
17
18
-shared uint sharedPosition;
19
shared BatchInfo sharedInfo;
20
21
void main() {
22
const uint threadIndex = gl_LocalInvocationID.x;
23
const uint batchIndex = gl_GlobalInvocationID.y;
24
25
if (threadIndex == 0) {
26
- sharedPosition = uint(x[batchIndex]);
27
sharedInfo = infos[batchIndex];
28
}
29
30
barrier();
31
- memoryBarrierShared();
32
33
const uint inputSizeX = sharedInfo.inputSizeX;
34
const uint xOffset = sharedInfo.inputOffset;
0 commit comments