vllm/distributed/device_communicators/symm_mem.py · 88f1b374f53e1d079dda26966dfd4422cf84a978 · 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

symm_mem.py 5.28 KB