Commit c16dd0a
committed
[ET-VK][reduction] Work around Adreno 8x4 dispatch failure
## Ulterior Motive
Restore ET-VK operator correctness on Galaxy S25 devices used by Eureka CI.
## Rationale
**What**: Make `reduce_nworkers()` select four workers when scaling would
select eight.
**Why**: Galaxy S25's Adreno Vulkan driver skips alternating work groups for
the resulting `8x4` local size. Reduction outputs retain uninitialized data
for those groups.
## Details
| Behavior | Before | After |
| --- | --- | --- |
| Short reductions selecting eight workers | Dispatch `8x4`; eight generated tests fail | Dispatch `4x4`; all generated tests pass |
| Larger reductions | Scale to 16, 32, or 64 workers | Unchanged |
Authored with Codex.
Differential Revision: [D122891447](https://our.internmc.facebook.com/intern/diff/D122891447/)
ghstack-source-id: 440132327
Pull-Request: pytorch#233251 parent fdd5140 commit c16dd0a
1 file changed
Lines changed: 3 additions & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
119 | 119 | | |
120 | 120 | | |
121 | 121 | | |
122 | | - | |
| 122 | + | |
| 123 | + | |
| 124 | + | |
123 | 125 | | |
124 | 126 | | |
125 | 127 | | |
| |||
0 commit comments