tests/kernels/test_pos_encoding.py · 72d3a30c6327e70de3595d00f04e2d577fcbbb68 · SIYIXNI / vllm · GitLab

"example/vscode:/vscode.git/clone" did not exist on "1ddc3ec736c65c4e1bbf12cb6eb5138b123e01dc"

Find file Blame History Permalink

Remove hardcoded `device="cuda" ` to support more devices (#2503) · 96b6f475
Kunshang Ji authored Feb 02, 2024
```
Co-authored-by: Jiang Li <jiang1.li@intel.com>
Co-authored-by: Kunshang Ji <kunshang.ji@intel.com>
```
96b6f475

test_pos_encoding.py 2.3 KB