tests/kernels/test_prefix_prefill.py · 8b430d7dea5695324636fc458c1cce52213bd499 · norm / vllm · GitLab

"configs/datasets/CLUE_CMRC/CLUE_CMRC_gen_3749cd.py" did not exist on "7d346000bb8f1f7611f88dc8e003bdf8c9ae3ece"

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_prefix_prefill.py 5.82 KB