tests/worker/spec_decode/utils.py · 7e1d5e5308fa3549dfed1821188d588260a03c8a · kecinstone / 2024pra-vllm · GitLab

"transformer_engine/jax/layernorm_mlp.py" did not exist on "cd798c9713c3d9f7bc091e679f51a7a2f44eef10"

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

utils.py 5.8 KB