vllm/v1/worker/gpu_ubatch_wrapper.py · 354cd580d53abd9b1b5896afc8a9dba61a9063df · OpenDAS / vllm_cscc · GitLab

Find file Blame History Permalink

[Hardware] Replace torch.cuda.device_count/current_device/set_device API (#36145) · 53ec16a7
Kunshang Ji authored Mar 12, 2026
```
Signed-off-by: Kunshang Ji <jikunshang95@gmail.com>
Signed-off-by: Kunshang Ji <kunshang.ji@intel.com>
```
53ec16a7

gpu_ubatch_wrapper.py 19.3 KB