tests/distributed/test_pynccl.py · 3fb4b5fa7442f98d864a0563e50837292172a713 · 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

test_pynccl.py 14.1 KB