[Core][Bugfix] Use correct device to initialize GPU data during CUDA-graph-capture

[yanburman]

Until now, tensors and tiles were created over the default device. When trying to capture a CUDA-graph over other different devices, the code failed since the data was created on a different GPU.

This PR fixes this problem by initializing the data on the correct device.

[github-actions]

👋 Hi! Thank you for contributing to the vLLM project.
Just a reminder: PRs would not trigger full CI run by default. Instead, it would only run fastcheck CI which starts running only a small and essential subset of CI tests to quickly catch errors. You can run other CI tests on top of those by going to your fastcheck build on Buildkite UI (linked in the PR checks section) and unblock them. If you do not have permission to unblock, ping simon-mo or khluu to add you in our Buildkite org.

Once the PR is approved and ready to go, your PR reviewer(s) can run CI to test the changes comprehensively before merging.

To run CI, PR reviewers can do one of these:

• Add ready label to the PR
• Enable auto-merge.

🚀

[youkaichao]

makes sense to me. but right now we only use one GPU per process. What's the use case when you have another GPU here?

[yanburman]

makes sense to me. but right now we only use one GPU per process. What's the use case when you have another GPU here?

We are using vllm as a library and were hitting this. Sounds like a good practice to not assume running on single GPU, especially if it doesn't cost anything

[youkaichao]

thanks for the contribution! we can merge it as long as ci passes.