vllm

[Hardware] Replace torch.cuda.device_count/current_device/set_device API

#36145

Merged