"sgl-router/vscode:/vscode.git/clone" did not exist on "a578d300ba036221894113c7ec0d415a8dbb319e"
- 11 Jun, 2025 1 commit
-
-
Michael Yang authored
The current splitDim function only operates on tensors that are split evenly which isn't always the case, e.g. a QKV tensor. This change allows the function to be used for arbitrary splits
-