Commit c1e6b07
committed
Add narrow op dispatch to ComputeGraphExecutor
The executor was missing a case for the narrow operation, causing
graph execution to fail when MultiHeadAttention traces GQA head
expansion (repeatKVHeads) into the optimized compute graph.1 parent f3a0393 commit c1e6b07
1 file changed
Lines changed: 6 additions & 0 deletions
File tree
- skainet-compile/skainet-compile-dag/src/commonMain/kotlin/sk/ainet/lang/graph/exec
Lines changed: 6 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
195 | 195 | | |
196 | 196 | | |
197 | 197 | | |
| 198 | + | |
| 199 | + | |
| 200 | + | |
| 201 | + | |
| 202 | + | |
| 203 | + | |
198 | 204 | | |
199 | 205 | | |
200 | 206 | | |
| |||
0 commit comments