File tree Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -51,7 +51,9 @@ void main() {
51
51
const uint workGroupIndex = gl_WorkGroupID.x;
52
52
const uint batchIndex = gl_WorkGroupID.y;
53
53
const uint zIndex = gl_WorkGroupID.z;
54
- const BatchInfo info = infos[zIndex * N_BATCHES + batchIndex];
54
+
55
+ const uint b = zIndex * N_BATCHES + batchIndex;
56
+ const BatchInfo info = infos[b];
55
57
56
58
const uint expertIndex = nExperts == 0
57
59
? 0
@@ -60,7 +62,6 @@ void main() {
60
62
61
63
const uint inputOffset = info.inputOffset;
62
64
const uint inputSizeX = info.inputSizeX;
63
- const uint outputOffset = info.outputOffset;
64
65
const uint d = TILE_SIZE_D * workGroupIndex;
65
66
66
67
vec4 xTemp[Q80_Q40_BLOCK_SIZE / 4];
@@ -102,6 +103,8 @@ void main() {
102
103
103
104
barrier();
104
105
106
+ const uint outputOffset = infos[b].outputOffset; // Hoisting fix for Raspberry PI
107
+
105
108
uint i = N_THREADS;
106
109
while (i % 2 == 0) {
107
110
i >>= 1;
You can’t perform that action at this time.
0 commit comments